diff --git a/tests/pytests/unit/conftest.py b/tests/pytests/unit/conftest.py
index e19db6cf5a35..b3dc4bb32ebc 100644
--- a/tests/pytests/unit/conftest.py
+++ b/tests/pytests/unit/conftest.py
@@ -68,10 +68,10 @@ def syndic_opts(tmp_path):
 
 @pytest.fixture
 def mocked_tcp_pub_client():
-    transport = MagicMock(spec=salt.transport.tcp.TCPPubClient)
+    transport = MagicMock(spec=salt.transport.tcp.PublishClient)
     transport.connect = MagicMock()
     future = asyncio.Future()
     transport.connect.return_value = future
     future.set_result(True)
-    with patch("salt.transport.tcp.TCPPubClient", transport):
+    with patch("salt.transport.tcp.PublishClient", transport):
         yield