diff --git a/lib/gitlab/database/load_balancing/setup.rb b/lib/gitlab/database/load_balancing/setup.rb
index 4860bffb86c5692e45b46052afcf59c7da109050..6e26b5235c486cee0b722b016caa4be86f5e4f6b 100644
--- a/lib/gitlab/database/load_balancing/setup.rb
+++ b/lib/gitlab/database/load_balancing/setup.rb
@@ -53,7 +53,7 @@ def setup_connection_proxy
 
           @model.singleton_class.alias_method(:lease_connection, :connection)
           @model.singleton_class.define_method(:with_connection) do |*_args, **_kwargs, &block|
-            next block&.call(connection) unless connection.is_a?(ConnectionProxy)
+            next block&.call(connection) unless connection.is_a?(::Gitlab::Database::LoadBalancing::ConnectionProxy)
 
             connection_already_checked_out = load_balancer.connection_checked_out?