lib/flail/exception.rb in flail-0.0.2 vs lib/flail/exception.rb in flail-0.0.3

@@ -18,9 +18,10 @@
                    end
     end
 
     def controller
       @controller ||= @env['action_controller.instance']
+      @controller ||= @request.controller if @request.respond_to?(:controller)
     end
 
     def user
       if controller.respond_to?(:current_user)
         current_user = controller.current_user