diff --git a/activerecord/test/cases/adapters/postgresql/connection_test.rb b/activerecord/test/cases/adapters/postgresql/connection_test.rb
index 47089d5d0d..1ff307c735 100644
--- a/activerecord/test/cases/adapters/postgresql/connection_test.rb
+++ b/activerecord/test/cases/adapters/postgresql/connection_test.rb
@@ -92,9 +92,9 @@ module ActiveRecord
         def query_fake(*args)
           if !( @called ||= false )
             self.stubs(:status).returns(PGconn::CONNECTION_BAD)
+            @called = true
             raise PGError
           else
-            @called = true
             self.unstub(:status)
             query_unfake(*args)
           end