diff --git a/csharp/ql/test/library-tests/frameworks/EntityFramework/Dataflow.ql b/csharp/ql/test/library-tests/frameworks/EntityFramework/Dataflow.ql
index 343c8bdc354..34befb929ef 100644
--- a/csharp/ql/test/library-tests/frameworks/EntityFramework/Dataflow.ql
+++ b/csharp/ql/test/library-tests/frameworks/EntityFramework/Dataflow.ql
@@ -3,18 +3,18 @@
  */
 
 import csharp
-import DataFlow::PathGraph
+import Taint::PathGraph
 
-class MyConfiguration extends TaintTracking::Configuration {
-  MyConfiguration() { this = "EntityFramework dataflow" }
+module TaintConfig implements DataFlow::ConfigSig {
+  predicate isSource(DataFlow::Node node) { node.asExpr().getValue() = "tainted" }
 
-  override predicate isSource(DataFlow::Node node) { node.asExpr().getValue() = "tainted" }
-
-  override predicate isSink(DataFlow::Node node) {
+  predicate isSink(DataFlow::Node node) {
     node.asExpr() = any(MethodCall c | c.getTarget().hasName("Sink")).getAnArgument()
   }
 }
 
-from DataFlow::PathNode source, DataFlow::PathNode sink, MyConfiguration conf
-where conf.hasFlowPath(source, sink)
+module Taint = TaintTracking::Global<TaintConfig>;
+
+from Taint::PathNode source, Taint::PathNode sink
+where Taint::flowPath(source, sink)
 select sink, source, sink, "$@", source, source.toString()