diff --git a/app/assets/javascripts/protocols/steps.js.erb b/app/assets/javascripts/protocols/steps.js.erb
index beb7e43ba..8ab161fbe 100644
--- a/app/assets/javascripts/protocols/steps.js.erb
+++ b/app/assets/javascripts/protocols/steps.js.erb
@@ -184,8 +184,8 @@
       // Switch position of top and bottom steps
       if (!_.isUndefined($stepDown) && !_.isUndefined($stepUp)) {
         $stepDown.insertAfter($stepUp);
-        $stepDown.find(".badge").html(stepDownPosition+1);
-        $stepUp.find(".badge").html(stepUpPosition+1);
+        $stepDown.find(".step-number").html(stepDownPosition + 1);
+        $stepUp.find(".step-number").html(stepUpPosition + 1);
         $("html, body").animate({ scrollTop: $step.offset().top - window.innerHeight / 2 });
       }
 
@@ -220,7 +220,7 @@
     });
   }
 
-    // Init ajax success/error for edit form  
+    // Init ajax success/error for edit form
   function formEditAjax($form) {
     $form
     .on("ajax:beforeSend", function () {
@@ -249,7 +249,7 @@
       });
       setupAssetsLoading();
     })
-    .on("ajax:error", function(e, xhr, status, error) { 
+    .on("ajax:error", function(e, xhr, status, error) {
       $form.renderFormErrors('step', xhr.responseJSON, true, e);
 
       formCallback($form);