diff --git a/python/ql/test/experimental/dataflow/coverage/classesCallGraph.ql b/python/ql/test/experimental/dataflow/coverage/classesCallGraph.ql
index 88febf44060..c1b66d0f323 100644
--- a/python/ql/test/experimental/dataflow/coverage/classesCallGraph.ql
+++ b/python/ql/test/experimental/dataflow/coverage/classesCallGraph.ql
@@ -20,7 +20,7 @@ class CallGraphConfig extends DataFlow::Configuration {
     node instanceof DataFlow::ParameterNode and
     // exclude parameters to the SINK-functions
     not exists(DataFlowPrivate::DataFlowCallable c |
-      node.(DataFlow::SourceParameterNode).isParameterOf(c, _) and
+      c.getParameter(_) = node.asCfgNode() and
       c.getName().matches("SINK_")
     )
   }