diff --git a/python/distrdf/common/check_rungraphs.py b/python/distrdf/common/check_rungraphs.py
index 4fb9009e72..b6d3c85f0b 100644
--- a/python/distrdf/common/check_rungraphs.py
+++ b/python/distrdf/common/check_rungraphs.py
@@ -4,7 +4,6 @@
 
 import ROOT
 
-import DistRDF
 from DistRDF.Backends import Spark
 from DistRDF.Backends import Dask
 
@@ -46,7 +45,7 @@ def test_rungraphs_sparkanddask_3histos(self, connection):
         for proxy in histoproxies:
             assert proxy.proxied_node.value is None
 
-        DistRDF.RunGraphs(histoproxies)
+        ROOT.RDF.RunGraphs(histoproxies)
 
         # After RunGraphs all histograms are correctly assigned to the
         # node objects
diff --git a/python/distrdf/dask/check_rungraphs.py b/python/distrdf/dask/check_rungraphs.py
index 5fe348992d..82ffe232b8 100644
--- a/python/distrdf/dask/check_rungraphs.py
+++ b/python/distrdf/dask/check_rungraphs.py
@@ -4,7 +4,6 @@
 
 import ROOT
 
-import DistRDF
 from DistRDF.Backends import Dask
 
 
@@ -36,7 +35,7 @@ def test_rungraphs_dask_3histos(self, connection):
         for proxy in histoproxies:
             assert proxy.proxied_node.value is None
 
-        DistRDF.RunGraphs(histoproxies)
+        ROOT.RDF.RunGraphs(histoproxies)
 
         # After RunGraphs all histograms are correctly assigned to the
         # node objects
diff --git a/python/distrdf/spark/check_rungraphs.py b/python/distrdf/spark/check_rungraphs.py
index b187084e16..5ff6c49d6b 100644
--- a/python/distrdf/spark/check_rungraphs.py
+++ b/python/distrdf/spark/check_rungraphs.py
@@ -4,7 +4,6 @@
 
 import ROOT
 
-import DistRDF
 from DistRDF.Backends import Spark
 
 
@@ -36,7 +35,7 @@ def test_rungraphs_spark_3histos(self, connection):
         for proxy in histoproxies:
             assert proxy.proxied_node.value is None
 
-        DistRDF.RunGraphs(histoproxies)
+        ROOT.RDF.RunGraphs(histoproxies)
 
         # After RunGraphs all histograms are correctly assigned to the
         # node objects