diff --git a/tests/test_zero_data_parallel/test_sharded_optim_with_sync_bn.py b/tests/test_zero_data_parallel/test_sharded_optim_with_sync_bn.py
index a0667792f..738382d8c 100644
--- a/tests/test_zero_data_parallel/test_sharded_optim_with_sync_bn.py
+++ b/tests/test_zero_data_parallel/test_sharded_optim_with_sync_bn.py
@@ -4,7 +4,6 @@
 from functools import partial
 
 import colossalai
-import pyte
 import pytest
 import torch
 import torch.distributed as dist