diff --git a/codes/utils/util.py b/codes/utils/util.py
index 2ae60bb2..cf904807 100644
--- a/codes/utils/util.py
+++ b/codes/utils/util.py
@@ -15,7 +15,7 @@ from shutil import get_terminal_size
 import scp
 import paramiko
 import options.options as options
-from utils.util import checkpoint
+from torch.utils.checkpoint import checkpoint
 
 import yaml
 try: