Update checking safe mode notification.

Signed-off-by: crynobone <crynobone@gmail.com>
orchestral · Feb 11, 2015 · 93f8ef5 · 93f8ef5
1 parent fd7dcd1
commit 93f8ef5
Showing 1 changed file with 5 additions and 7 deletions.
diff --git a/src/Bootstrap/NotifyIfSafeMode.php b/src/Bootstrap/NotifyIfSafeMode.php
@@ -12,12 +12,10 @@ class NotifyIfSafeMode
      */
     public function bootstrap(Application $app)
     {
-        $app['events']->listen('orchestra.extension: booted', function () use ($app) {
-            if ('on' === $app['session']->get('orchestra.safemode')) {
-                $app['orchestra.messages']->extend(function ($messages) {
-                    $messages->add('info', trans('orchestra/foundation::response.safe-mode'));
-                });
-            }
-        });
+        if ($app['orchestra.extension.mode']->check()) {
+            $app['orchestra.messages']->extend(function ($messages) {
+                $messages->add('info', trans('orchestra/foundation::response.safe-mode'));
+            });
+        }
     }
 }