diff --git a/force_bdss/io/tests/test_workflow_writer.py b/force_bdss/io/tests/test_workflow_writer.py
index af25d561258b5c101dbd2d57506a446c949f2017..e6f81ec25a52ff167f25109b4dc37d3c43f42798 100644
--- a/force_bdss/io/tests/test_workflow_writer.py
+++ b/force_bdss/io/tests/test_workflow_writer.py
@@ -28,7 +28,7 @@ class TestWorkflowWriter(unittest.TestCase):
             bundle=mock_mco_bundle
         )
         mock_mco_bundle.create_model = mock.Mock(
-            return_value = mock_mco_model
+            return_value=mock_mco_model
         )
         self.mock_registry.mco_bundle_by_id = mock.Mock(
             return_value=mock_mco_bundle)
diff --git a/force_bdss/io/workflow_writer.py b/force_bdss/io/workflow_writer.py
index d0b2c8d3de3f8676c6d9bb85ea091167937eb6d9..129e64724d8d685f90028952c03d6d785315f04f 100644
--- a/force_bdss/io/workflow_writer.py
+++ b/force_bdss/io/workflow_writer.py
@@ -1,7 +1,5 @@
 import json
-from traits.api import Instance, HasStrictTraits
-
-from ..bundle_registry_plugin import BundleRegistryPlugin
+from traits.api import HasStrictTraits
 
 
 class WorkflowWriter(HasStrictTraits):