diff --git a/app/views/problems/toggle.js.haml b/app/views/problems/toggle.js.haml
--- a/app/views/problems/toggle.js.haml
+++ b/app/views/problems/toggle.js.haml
@@ -1,10 +1,6 @@
+= render partial: 'toggle_button',
+                  locals: {button_id: "#problem-avail-#{@problem.id}",button_on: @problem.available }
 :plain
-  b = $("#prob-#{@problem.id}-avail");
-  b.removeClass('btn-default');
-  b.removeClass('btn-success');
-  b.removeClass('btn-warning');
-  b.addClass("btn-#{@problem.available? ? 'success' : 'default'}");
-  b.text("#{@problem.available? ? 'Yes' : 'No'}");
   r = $("#prob-#{@problem.id}");
   r.removeClass('success');
   r.removeClass('danger');