diff --git a/deepspeed/runtime/bf16_optimizer.py b/deepspeed/runtime/bf16_optimizer.py
index 303267f0494d..a9988e2c498f 100644
--- a/deepspeed/runtime/bf16_optimizer.py
+++ b/deepspeed/runtime/bf16_optimizer.py
@@ -2,7 +2,7 @@
 Copyright 2022 The Microsoft DeepSpeed Team
 """
 
-from typing import OrderedDict
+from collections import OrderedDict
 import torch
 import os
 from deepspeed import comm as dist