diff --git a/tests/models/hf/test_hf_base.py b/tests/models/hf/test_hf_base.py
index bd1ac6f8a8..ec45e31a65 100644
--- a/tests/models/hf/test_hf_base.py
+++ b/tests/models/hf/test_hf_base.py
@@ -2,9 +2,10 @@
 # SPDX-License-Identifier: Apache-2.0
 
 import pytest
-from llmfoundry.models.hf.hf_base import BaseHuggingFaceModel
 from peft import PeftModel
 
+from llmfoundry.models.hf.hf_base import BaseHuggingFaceModel
+
 
 def test_build_inner_model_fsdp():
     model = BaseHuggingFaceModel.build_inner_model(
@@ -45,4 +46,4 @@ def test_pretrained_peft_trainable():
 
     n_trainable, n_all = model.get_nb_trainable_parameters()
     assert n_all > 0
-    assert n_trainable > 0
\ No newline at end of file
+    assert n_trainable > 0