diff --git a/test/integration/agents/test_ri_communication_agent.py b/test/integration/agents/test_ri_communication_agent.py
index 33051c8..a82a837 100644
--- a/test/integration/agents/test_ri_communication_agent.py
+++ b/test/integration/agents/test_ri_communication_agent.py
@@ -428,6 +428,7 @@ async def test_listen_behaviour_ping_correct(caplog):
     # TODO: Integration test between actual server and password needed for spade agents
     agent = RICommunicationAgent("test@server", "password")
     agent._req_socket = fake_socket
+    agent.connected = True
 
     behaviour = agent.ListenBehaviour()
     agent.add_behaviour(behaviour)
@@ -463,6 +464,7 @@ async def test_listen_behaviour_ping_wrong_endpoint(caplog):
 
     agent = RICommunicationAgent("test@server", "password", fake_pub_socket)
     agent._req_socket = fake_socket
+    agent.connected = True
 
     behaviour = agent.ListenBehaviour()
     agent.add_behaviour(behaviour)
@@ -486,6 +488,7 @@ async def test_listen_behaviour_timeout(zmq_context, caplog):
 
     agent = RICommunicationAgent("test@server", "password")
     agent._req_socket = fake_socket
+    agent.connected = True
 
     behaviour = agent.ListenBehaviour()
     agent.add_behaviour(behaviour)
@@ -514,6 +517,7 @@ async def test_listen_behaviour_ping_no_endpoint(caplog):
 
     agent = RICommunicationAgent("test@server", "password")
     agent._req_socket = fake_socket
+    agent.connected = True
 
     behaviour = agent.ListenBehaviour()
     agent.add_behaviour(behaviour)