diff --git a/utils/adabound.py b/utils/adabound.py
index 142b1262..8baa3780 100644
--- a/utils/adabound.py
+++ b/utils/adabound.py
@@ -1,7 +1,7 @@
 import math
 
 import torch
-from torch.optim import Optimizer
+from torch.optim.optimizer import Optimizer
 
 
 class AdaBound(Optimizer):