diff --git a/optimizers/test_optimizer.py b/optimizers/test_optimizer.py
index 5ef1107..e396c41 100644
--- a/optimizers/test_optimizer.py
+++ b/optimizers/test_optimizer.py
@@ -1,7 +1,7 @@
 import unittest
 import torch
 from tml.optimizers.config import LearningRate, OptimizerConfig
-from .optimizer import compute_lr, LRShim, get_optimizer_class, build_optimizer
+from optimizer import compute_lr, LRShim, get_optimizer_class, build_optimizer
 
 
 class TestComputeLR(unittest.TestCase):