diff --git a/python/ql/lib/semmle/python/frameworks/data/internal/ApiGraphModelsSpecific.qll b/python/ql/lib/semmle/python/frameworks/data/internal/ApiGraphModelsSpecific.qll
index 206a6ff1e52..ae1c59a30d0 100644
--- a/python/ql/lib/semmle/python/frameworks/data/internal/ApiGraphModelsSpecific.qll
+++ b/python/ql/lib/semmle/python/frameworks/data/internal/ApiGraphModelsSpecific.qll
@@ -20,9 +20,9 @@
  */
 
 private import python as PY
-import semmle.python.dataflow.new.DataFlow
+private import semmle.python.dataflow.new.DataFlow
 private import ApiGraphModels
-import semmle.python.ApiGraphs
+import semmle.python.ApiGraphs::API as API
 
 class Unit = PY::Unit;