diff --git a/fastreid/data/build.py b/fastreid/data/build.py
index 23ac8d6..681bc16 100644
--- a/fastreid/data/build.py
+++ b/fastreid/data/build.py
@@ -16,7 +16,7 @@ if TORCH_MAJOR == 1 and TORCH_MINOR < 8:
 else:
     string_classes = str
 
-from collections import Mapping
+from collections.abc import Mapping
 
 from fastreid.config import configurable
 from fastreid.utils import comm
diff --git a/fastreid/evaluation/testing.py b/fastreid/evaluation/testing.py
index cf4abc3..5fe82c9 100644
--- a/fastreid/evaluation/testing.py
+++ b/fastreid/evaluation/testing.py
@@ -2,7 +2,8 @@
 import logging
 import pprint
 import sys
-from collections import Mapping, OrderedDict
+from collections import OrderedDict
+from collections.abc import Mapping
 
 import numpy as np
 from tabulate import tabulate