diff --git a/dags/dynamic_data_workflow_dag.py b/dags/dynamic_data_workflow_dag.py
index f5145267..b71ad89d 100644
--- a/dags/dynamic_data_workflow_dag.py
+++ b/dags/dynamic_data_workflow_dag.py
@@ -17,7 +17,7 @@
     container_definitions=[
         {
             "name": "collection-task",
-            "image": "public.ecr.aws/l6z6v3j6/development-mwaa-dataset-collection-task",
+            "image": "public.ecr.aws/l6z6v3j6/development-mwaa-dataset-collection-task:publish-image",
             # "workingDirectory": "/usr/bin",
             # "entryPoint": ["sh", "-c"],
             # "command": ["ls"],