lib/assert-rack-test.rb in assert-rack-test-1.0.0 vs lib/assert-rack-test.rb in assert-rack-test-1.0.1

@@ -1,14 +1,22 @@
 require 'assert'
 require 'rack/test'
 
+require 'assert-rack-test/version'
+
 module Assert
   module Rack
     module Test
-      autoload :VERSION, "assert-rack-test/version"
+
+      def self.included(klass)
+        klass.class_eval do
+          include ::Rack::Test::Methods
+        end
+      end
+
     end
   end
 end
 
 Assert::Context.class_eval do
-  include Rack::Test::Methods
+  include Assert::Rack::Test
 end