diff --git a/tests/unit/alexnet_model.py b/tests/unit/alexnet_model.py
index c9a985934047..cf533063d6ec 100644
--- a/tests/unit/alexnet_model.py
+++ b/tests/unit/alexnet_model.py
@@ -10,7 +10,8 @@
 import torch.nn.functional as F
 import deepspeed
 import deepspeed.comm as dist
-import deepspeed.runtime.utils as ds_utils, required_torch_version
+import deepspeed.runtime.utils as ds_utils
+from deepspeed.runtime.utils import required_torch_version
 from deepspeed.accelerator import get_accelerator
 from deepspeed.runtime.pipe.module import PipelineModule, LayerSpec