diff --git a/paddleslim/common/controller_server.py b/paddleslim/common/controller_server.py
index 10883d5988652a8c3f738416eae3dd768ba74e67..74b954db3bb1c4520551e82b5e8ba3b9514c549c 100644
--- a/paddleslim/common/controller_server.py
+++ b/paddleslim/common/controller_server.py
@@ -17,7 +17,7 @@ import logging
 import socket
 from .log_helper import get_logger
 from threading import Thread
-from .lock import lock, unlock
+from .lock_utils import lock, unlock
 
 __all__ = ['ControllerServer']