diff --git a/lib/galaxy/workflow/modules.py b/lib/galaxy/workflow/modules.py
index 4433d6b8b9d..f1c5c2da996 100644
--- a/lib/galaxy/workflow/modules.py
+++ b/lib/galaxy/workflow/modules.py
@@ -1145,6 +1145,7 @@ class ToolModule( WorkflowModule ):
                         if isinstance( input, DataToolParameter ):
                             # Pull out dataset instance from element.
                             replacement = iteration_elements[ prefixed_name ].dataset_instance
+                            replacement.element_identifier = iteration_elements[ prefixed_name ].element_identifier
                         else:
                             # If collection - just use element model object.
                             replacement = iteration_elements[ prefixed_name ]