diff --git a/python/lsst/ctrl/mpexec/__init__.py b/python/lsst/ctrl/mpexec/__init__.py
index e37e78bf..2405f748 100644
--- a/python/lsst/ctrl/mpexec/__init__.py
+++ b/python/lsst/ctrl/mpexec/__init__.py
@@ -26,7 +26,6 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 
 from .cmdLineFwk import *
-from .dotTools import *
 from .executionGraphFixup import *
 from .mpGraphExecutor import *
 from .preExecInit import *
diff --git a/python/lsst/ctrl/mpexec/cmdLineFwk.py b/python/lsst/ctrl/mpexec/cmdLineFwk.py
index 7d9f0bcf..90654e09 100644
--- a/python/lsst/ctrl/mpexec/cmdLineFwk.py
+++ b/python/lsst/ctrl/mpexec/cmdLineFwk.py
@@ -68,6 +68,7 @@
     QuantumGraph,
     TaskFactory,
     buildExecutionButler,
+    graph2dot,
 )
 from lsst.pipe.base.all_dimensions_quantum_graph_builder import AllDimensionsQuantumGraphBuilder
 from lsst.pipe.base.pipeline_graph import NodeType
@@ -75,7 +76,6 @@
 from lsst.utils.logging import getLogger
 from lsst.utils.threads import disable_implicit_threading
 
-from .dotTools import graph2dot
 from .executionGraphFixup import ExecutionGraphFixup
 from .mpGraphExecutor import MPGraphExecutor
 from .preExecInit import PreExecInit, PreExecInitLimited
@@ -583,7 +583,10 @@ def makePipeline(self, args: SimpleNamespace) -> Pipeline:
             pipeline.write_to_uri(args.save_pipeline)
 
         if args.expand_pipeline:
-            pipeline.write_to_uri(args.expand_pipeline, expand=True)
+            task_defs = list(pipeline)
+            pipeline.write_to_uri(args.expand_pipeline, expand=True, task_defs=task_defs)
+        else:
+            task_defs = None
 
         return pipeline