diff --git a/tests/workflow/test_workflow_service.py b/tests/workflow/test_workflow_service.py
index a4af4edc..dc6eb01e 100644
--- a/tests/workflow/test_workflow_service.py
+++ b/tests/workflow/test_workflow_service.py
@@ -10,6 +10,7 @@ from example_data import ExampleDataLoader
 from crc import db
 from crc.models.task_event import TaskEventModel
 from crc.models.api_models import Task
+from crc.models.file import FileModel
 from crc.api.common import ApiError
 
 
@@ -114,3 +115,12 @@ class TestWorkflowService(BaseTest):
 
         result2 = WorkflowService.get_dot_value(path, {"a.b.c":"garbage"})
         self.assertEqual("garbage", result2)
+
+    def test_get_primary_workflow(self):
+
+        workflow = self.create_workflow('hello_world')
+        workflow_spec_id = workflow.workflow_spec.id
+        primary_workflow = WorkflowService.get_primary_from_workflow_spec(workflow_spec_id)
+        self.assertIsInstance(primary_workflow, FileModel)
+        self.assertEqual(workflow_spec_id, primary_workflow.workflow_spec_id)
+        self.assertEqual('hello_world.bpmn', primary_workflow.name)