diff --git a/sleekxmpp/test/sleektest.py b/sleekxmpp/test/sleektest.py
index 9b822343..fa53bf73 100644
--- a/sleekxmpp/test/sleektest.py
+++ b/sleekxmpp/test/sleektest.py
@@ -281,9 +281,9 @@ class SleekTest(unittest.TestCase):
             self.xmpp.set_socket(TestSocket())
 
             # Simulate connecting for mock sockets.
-            self.xmpp.state.set('reconnect', False)
-            self.xmpp.state.set('is client', True)
-            self.xmpp.state.set('connected', True)
+            self.xmpp.auto_reconnect = False
+            self.xmpp.is_client = True
+            self.xmpp.state._set_state('connected')
 
             # Must have the stream header ready for xmpp.process() to work.
             if not header: