diff --git a/bioflow_insight_cli/main.py b/bioflow_insight_cli/main.py
index be9fdb451685a01b205926d32d79eaca953cd994..1ce525fdc8b4e26e372b63e68b91b2606d5a0363 100644
--- a/bioflow_insight_cli/main.py
+++ b/bioflow_insight_cli/main.py
@@ -1,6 +1,6 @@
 import click
 
-from src.nextflow_file import Nextflow_File
+from src.workflow import Workflow
 
 
 @click.command()
@@ -36,7 +36,7 @@ def cli(main_workflow_path, render_graphs: bool, **kwargs):
     in folders with eponymous names.
     """
 
-    w = Nextflow_File(address=main_workflow_path, **kwargs)
+    w = Workflow(address=main_workflow_path, **kwargs)
     w.initialise()
     w.generate_all_graphs(render_graphs=render_graphs)