@@ -1228,7 +1228,7 @@ def test_multi_database_polymorphic_preload_with_same_table_name
     # Expected
     #   SELECT FROM dogs ... (Dog)
     #   SELECT FROM dogs ... (OtherDog)
-    assert_queries(2) do
+    assert_queries_count(2) do
       preloader = ActiveRecord::Associations::Preloader.new(records: [dog_comment, other_dog_comment], associations: :origin)
       preloader.call
     end