diff --git a/examples/optimizer.py b/examples/optimizer.py
index 8e5ef065..396062bc 100644
--- a/examples/optimizer.py
+++ b/examples/optimizer.py
@@ -1,5 +1,5 @@
-from torch.optim import SGD, Adam
-from transformers import AdamW
+from torch.optim import SGD, Adam, AdamW
+
 
 def initialize_optimizer(config, model):
     # initialize optimizers