diff --git a/tests/test_module.py b/tests/test_module.py
index d73dea5..07fbb18 100644
--- a/tests/test_module.py
+++ b/tests/test_module.py
@@ -7,7 +7,9 @@
 import pytest
 import requests
 from wei import ExperimentClient
-from wei.core.data_classes import ModuleAbout, WorkcellData, WorkflowStatus
+from wei.types.module_types import ModuleAbout
+from wei.types.workcell_types import WorkcellData
+from wei.types.workflow_types import WorkflowStatus
 
 
 class TestWEI_Base(unittest.TestCase):