diff --git a/test/multi_tenancy_test.rb b/test/multi_tenancy_test.rb
index d66d024..0fb3870 100644
--- a/test/multi_tenancy_test.rb
+++ b/test/multi_tenancy_test.rb
@@ -7,16 +7,20 @@ class MultiTenancyTest < Minitest::Test
 
   def test_basic
     Apartment::Tenant.switch!("tenant1")
-    store_names ["Product A"], Tenant
+    store_names ["Product A"]
     Apartment::Tenant.switch!("tenant2")
-    store_names ["Product B"], Tenant
+    store_names ["Product B"]
     Apartment::Tenant.switch!("tenant1")
-    assert_search "product", ["Product A"], {load: false}, Tenant
+    assert_search "product", ["Product A"], {load: false}
     Apartment::Tenant.switch!("tenant2")
-    assert_search "product", ["Product B"], {load: false}, Tenant
+    assert_search "product", ["Product B"], {load: false}
   end
 
   def teardown
     Apartment::Tenant.reset if defined?(Apartment)
   end
+
+  def default_model
+    Tenant
+  end
 end
--
libgit2 0.21.0