lib/lookout/rack/test/cucumber/server.rb in lookout-rack-test-1.1.0 vs lib/lookout/rack/test/cucumber/server.rb in lookout-rack-test-1.2.0

@@ -1,10 +1,14 @@
 require 'rack/test'
+require 'lookout/rack/test'
 
 module Lookout::Rack::Test::Cucumber
   module Server
     def app
-      @app ||= $application_class.new if $application_class
+      return @app if @app
+
+      klass = ::Lookout::Rack::Test.app || $application_class
+      @app = klass.new if klass
     end
   end
 end
 
 World(Lookout::Rack::Test::Cucumber)