diff --git a/CODEOWNERS b/CODEOWNERS
index 3eb8710cad7a..334cd55db205 100644
--- a/CODEOWNERS
+++ b/CODEOWNERS
@@ -1 +1 @@
-*       @jeffra @samyam @tjruwase @ShadenSmith @conglongli @awan-10 @cli99 @eltonzheng @minjiaz @RezaYazdaniAminabadi @duli2012 @mrwyattii @yaozhewei @arashb @xiaoxiawu-microsoft
+*       @jeffra @samyam @tjruwase @ShadenSmith @conglongli @awan-10 @cli99 @eltonzheng @minjiaz @RezaYazdaniAminabadi @duli2012 @mrwyattii @yaozhewei @arashb @xiaoxiawu-microsoft @samadejacobs
diff --git a/tests/small_model_debugging/test_model.py b/tests/small_model_debugging/test_model.py
index 720adeab3842..ec3325e24add 100755
--- a/tests/small_model_debugging/test_model.py
+++ b/tests/small_model_debugging/test_model.py
@@ -4,6 +4,7 @@
 import torch
 import deepspeed
 from torch.utils.data.distributed import DistributedSampler
+import deepspeed.comm as dist
 
 
 class SimpleModel(torch.nn.Module):