diff --git a/spec/support/kneet_connections.rb b/spec/support/kneet_connections.rb
index 7de8cc59..99c91fd3 100644
--- a/spec/support/kneet_connections.rb
+++ b/spec/support/kneet_connections.rb
@@ -1,9 +1,11 @@
-class ActiveRecord::Base
-  mattr_accessor :shared_connection
-  @@shared_connection = nil
+module ActiveRecord
+  class Base
+    mattr_accessor :shared_connection
+    @@shared_connection = nil
 
-  def self.connection
-    @@shared_connection || retrieve_connection
+    def self.connection
+      @@shared_connection || retrieve_connection
+    end
   end
 end