app/helpers/archangel/flash_helper.rb in archangel-0.0.5 vs app/helpers/archangel/flash_helper.rb in archangel-0.0.6

@@ -14,9 +14,9 @@
     def flash_class_for(flash_type)
       flash_type = flash_type.to_s.downcase.parameterize
 
       {
         success: "success", error: "danger", alert: "warning", notice: "info"
-      }[flash_type.to_sym] || flash_type
+      }.fetch(flash_type.to_sym, flash_type)
     end
   end
 end