diff --git a/.github/workflows/python-package.yml b/.github/workflows/python-package.yml
index 9e4c6f6..f12c64e 100644
--- a/.github/workflows/python-package.yml
+++ b/.github/workflows/python-package.yml
@@ -45,9 +45,9 @@ jobs:
         pip install dist/*.whl --force-reinstall
     - name: Test with pytest (TQ_ZERO_COPY_SERIALIZATION=False)
       run: |
-        pytest
+        pytest tests
     - name: Test with pytest (TQ_ZERO_COPY_SERIALIZATION=True)
       run: |
         ray stop --force
         export TQ_ZERO_COPY_SERIALIZATION=True
-        pytest
\ No newline at end of file
+        pytest tests
\ No newline at end of file
diff --git a/README.md b/README.md
index f87ccc0..7f46346 100644
--- a/README.md
+++ b/README.md
@@ -4,6 +4,7 @@
   </h2>
   <a href="https://arxiv.org/abs/2507.01663" target="_blank"><strong>Paper</strong></a>
 | <a href="https://zhuanlan.zhihu.com/p/1930244241625449814" target="_blank"><strong>Zhihu</strong></a>
+| <a href="https://github.com/TransferQueue/community_doc/blob/main/other_assets/WeChatGroup.png?raw=true" target="_blank"><strong>WeChat</strong></a>
   <br />
   <br />
 
diff --git a/scripts/performance_test.py b/scripts/performance_test.py
index 9586b40..396ddb2 100644
--- a/scripts/performance_test.py
+++ b/scripts/performance_test.py
@@ -32,8 +32,8 @@
 
 
 from transfer_queue import (  # noqa: E402
-    AsyncTransferQueueClient,
     SimpleStorageUnit,
+    TransferQueueClient,
     TransferQueueController,
     process_zmq_server_info,
 )
@@ -247,7 +247,7 @@ def _initialize_data_system(self):
         tq_config.storage_unit_infos = self.data_system_storage_unit_infos
         self.config = OmegaConf.merge(tq_config, self.config)
 
-        self.data_system_client = AsyncTransferQueueClient(
+        self.data_system_client = TransferQueueClient(
             client_id="Trainer", controller_info=self.data_system_controller_info
         )
         self.data_system_client.initialize_storage_manager(manager_type="AsyncSimpleStorageManager", config=self.config)