lib/boom_nats/application.rb in boom_nats-0.1.1 vs lib/boom_nats/application.rb in boom_nats-0.1.2

@@ -41,16 +41,14 @@
             end
           end
 
           BoomNats.logger.debug "BoomNats::started"
 
-          return if defined?(Rails::Railtie)
-
-          prepare_trap
+          prepare_trap unless defined?(Rails::Railtie)
         end
       end
 
-      wait
+      wait unless defined?(Rails::Railtie)
     end
 
     def stop
       @subscriptions.each { |s| nats.unsubscribe(s) }
       @subscriptions = []