diff --git a/tests/test_module.py b/tests/test_module.py
index 07fbb18..442033e 100644
--- a/tests/test_module.py
+++ b/tests/test_module.py
@@ -8,7 +8,7 @@
 import requests
 from wei import ExperimentClient
 from wei.types.module_types import ModuleAbout
-from wei.types.workcell_types import WorkcellData
+from wei.types.workcell_types import Workcell
 from wei.types.workflow_types import WorkflowStatus
 
 
@@ -22,7 +22,7 @@ def __init__(self, *args, **kwargs):
         self.workcell_file = self.root_dir / Path(
             "tests/workcell_defs/test_workcell.yaml"
         )
-        self.workcell = WorkcellData.from_yaml(self.workcell_file)
+        self.workcell = Workcell.from_yaml(self.workcell_file)
         self.server_host = self.workcell.config.server_host
         self.server_port = self.workcell.config.server_port
         self.url = f"http://{self.server_host}:{self.server_port}"