diff --git a/fastreid/utils/checkpoint.py b/fastreid/utils/checkpoint.py
index b01036c..7c32b1a 100644
--- a/fastreid/utils/checkpoint.py
+++ b/fastreid/utils/checkpoint.py
@@ -11,11 +11,11 @@ from typing import Optional, List, Dict, NamedTuple, Tuple, Iterable
 import numpy as np
 import torch
 import torch.nn as nn
+from apex.parallel import DistributedDataParallel
 from termcolor import colored
-from torch.nn.parallel import DataParallel, DistributedDataParallel
+from torch.nn.parallel import DataParallel
 
 from fastreid.utils.file_io import PathManager
-from fastreid.utils.events import get_event_storage
 
 
 class _IncompatibleKeys(