diff --git a/pungi_utils/orchestrator.py b/pungi_utils/orchestrator.py
index 3b8b26e8..d18019de 100644
--- a/pungi_utils/orchestrator.py
+++ b/pungi_utils/orchestrator.py
@@ -510,7 +510,7 @@ def run(work_dir, main_config_file, args):
         parts[section] = ComposePart.from_config(parser, section, config_dir)
 
     if hasattr(args, "part"):
-        setup_for_restart()
+        setup_for_restart(global_config, parts, args.part)
 
     return run_all(global_config, parts)