diff --git a/tests/collections/llm/auto_conf/test_base_configs.py b/tests/collections/llm/auto_conf/test_base_configs.py
index a120811330005..6cf3a3b63c76e 100644
--- a/tests/collections/llm/auto_conf/test_base_configs.py
+++ b/tests/collections/llm/auto_conf/test_base_configs.py
@@ -1,6 +1,4 @@
-from nemo.collections.llm.utils import Config, Partial
 import torch
-
 from megatron.core.optimizer import OptimizerConfig
 from pytorch_lightning.loggers import TensorBoardLogger
 
@@ -17,6 +15,7 @@
 )
 from nemo.collections.llm.tools.auto_configurator import AutoConfigurator
 from nemo.collections.llm.tools.auto_configurator.core.base_config import BaseConfig
+from nemo.collections.llm.utils import Config, Partial
 from nemo.lightning.pytorch.optim import CosineAnnealingScheduler, MegatronOptimizerModule
 from nemo.utils.exp_manager import TimingCallback
 
diff --git a/tests/collections/llm/auto_conf/test_generate_configs.py b/tests/collections/llm/auto_conf/test_generate_configs.py
index 9ca6ffe7c2c9c..99a1bf807244a 100644
--- a/tests/collections/llm/auto_conf/test_generate_configs.py
+++ b/tests/collections/llm/auto_conf/test_generate_configs.py
@@ -1,5 +1,3 @@
-from nemo.collections.llm.utils import Config, Partial
-
 from nemo.collections.llm import (
     GemmaConfig7B,
     GPTConfig5B,
@@ -9,6 +7,7 @@
     Nemotron3Config8B,
 )
 from nemo.collections.llm.tools.auto_configurator import AutoConfigurator, generate_configs
+from nemo.collections.llm.utils import Config, Partial
 
 
 def get_auto_configs(configs):