diff --git a/tests/test_optimizer_parameters.py b/tests/test_optimizer_parameters.py
index d9acd028..e7610e9e 100644
--- a/tests/test_optimizer_parameters.py
+++ b/tests/test_optimizer_parameters.py
@@ -2,16 +2,8 @@
 import torch
 from torch import nn
 
-from pytorch_optimizer.optimizer import (
-    SAM,
-    WSAM,
-    GaLoreProjector,
-    Lookahead,
-    PCGrad,
-    Ranger21,
-    SafeFP16Optimizer,
-    load_optimizer,
-)
+from pytorch_optimizer.optimizer import SAM, WSAM, Lookahead, PCGrad, Ranger21, SafeFP16Optimizer, load_optimizer
+from pytorch_optimizer.optimizer.galore import GaLoreProjector
 from tests.constants import PULLBACK_MOMENTUM
 from tests.utils import Example, simple_parameter, simple_zero_rank_parameter