diff --git a/monai/transforms/regularization/array.py b/monai/transforms/regularization/array.py
index e61740a9761..109fd414f49 100644
--- a/monai/transforms/regularization/array.py
+++ b/monai/transforms/regularization/array.py
@@ -11,9 +11,8 @@
 
 from abc import abstractmethod
 from typing import Tuple
-from monai.config import KeysCollection
 import torch
-from monai.transforms import Transform, MapTransform
+from monai.transforms import Transform
 from math import sqrt, ceil
 
 __all__ = ["MixUp", "CutMix", "CutOut"]