1 files changed, 12 insertions, 2 deletions
diff --git a/lib/helper.rb b/lib/helper.rb
index f4b05ad..9660f21 100644
--- a/lib/helper.rb
+++ b/lib/helper.rb
@@ -10,11 +10,21 @@ module Helper
   end
 
   def build_status_class build
-    if build.status == 'success'
+    if build.success?
       'label-success'
-    elsif build.status == 'fail'
+    elsif build.failed?
       'label-important'
     else
+      'label-inverse'
+    end
+  end
+
+  def build_status_alert_class build
+    if build.success?
+      'alert-success'
+    elsif build.failed?
+      'alert-error'
+    else
       ''
     end
   end