diff --git a/tests/unit/cli/test_generator.py b/tests/unit/cli/test_generator.py
index 6e9de2c1..187ec3c5 100644
--- a/tests/unit/cli/test_generator.py
+++ b/tests/unit/cli/test_generator.py
@@ -108,7 +108,7 @@ def test_generate_several_pullers_from_config(several_inputs_outputs_stream_conf
 
         elif current_puller_infos['type'] == 'socket':
             assert isinstance(db, SocketDB)
-            assert db.port == current_puller_infos['port']
+            assert db.server_address == (current_puller_infos['host'], current_puller_infos['port'])
 
         else:
             assert False