diff --git a/app/helpers/workflows_helper.rb b/app/helpers/workflows_helper.rb
index 80297ce12..8b040dafc 100644
--- a/app/helpers/workflows_helper.rb
+++ b/app/helpers/workflows_helper.rb
@@ -1,10 +1,10 @@
 module WorkflowsHelper
   def link_to_stage_picker(bubble, workflow)
-    link_to workflow.name, new_bucket_bubble_stage_picker_path(bubble.bucket, bubble, workflow_id: workflow.id)
+    link_to workflow.name, new_bucket_bubble_stage_picker_path(bubble.bucket, bubble, workflow_id: workflow)
   end
 
   def button_to_set_stage(bubble, stage)
-    button_to stage.name, bucket_bubble_stagings_path(bubble.bucket, bubble, stage_id: stage.id),
+    button_to stage.name, bucket_bubble_stagings_path(bubble.bucket, bubble, stage_id: stage),
       method: :post, class: [ "btn btn--small", { "fill-selected": stage == bubble.stage } ]
   end
 end