diff --git a/tests/UnitTests/WorkflowManager.Tests/Controllers/WorkflowsControllerTests.cs b/tests/UnitTests/WorkflowManager.Tests/Controllers/WorkflowsControllerTests.cs
index e5a96c706..387e6144d 100644
--- a/tests/UnitTests/WorkflowManager.Tests/Controllers/WorkflowsControllerTests.cs
+++ b/tests/UnitTests/WorkflowManager.Tests/Controllers/WorkflowsControllerTests.cs
@@ -237,7 +237,7 @@ public async Task UpdateAsync_WorkflowsDoesNotExist_ReturnsNotFound()
 
             var result = await WorkflowsController.UpdateAsync(newWorkflow, workflowRevision.WorkflowId);
 
-            var objectResult = Assert.IsType<NotFoundObjectResult>(result);
+            var objectResult = Assert.IsType<ObjectResult>(result);
 
             Assert.Equal(404, objectResult.StatusCode);
         }