diff --git a/core/tests/test_workflows.py b/core/tests/test_workflows.py
index 045fa8c5..22a57409 100644
--- a/core/tests/test_workflows.py
+++ b/core/tests/test_workflows.py
@@ -1,6 +1,5 @@
 from typing import Callable
 from copy import copy
-from importlib_resources import as_file
 from numpy.typing import NDArray
 
 from napari_workflows import Workflow
@@ -8,8 +7,6 @@
 
 from pandas import DataFrame
 from lls_core.models.lattice_data import LatticeData
-from lls_core.sample import resources
-from aicsimageio.aics_image import AICSImage
 
 from tests.utils import invoke
 from pathlib import Path
@@ -24,7 +21,7 @@ def test_napari_workflow(image_workflow: Workflow, test_image: NDArray):
     """
     workflow = copy(image_workflow)
     # Set input image to be the "raw" image
-    workflow.set("input", test_image)
+    workflow.set("deskewed_image", test_image)
     labeling = workflow.get("labeling")
     assert labeling[2, 2, 2] == 1
 
diff --git a/core/tests/workflows/argument_order/test_workflow.yml b/core/tests/workflows/argument_order/test_workflow.yml
index b74fb9f9..ba479d79 100644
--- a/core/tests/workflows/argument_order/test_workflow.yml
+++ b/core/tests/workflows/argument_order/test_workflow.yml
@@ -2,6 +2,6 @@
 _tasks:
   test: !!python/tuple
   - !!python/name:custom_function.test ''
-  - input_img
+  - deskewed_image
   - "abc"
-  - 3
\ No newline at end of file
+  - 3