diff --git a/tests/test_models.py b/tests/test_models.py
index 43ae5524..30d09e38 100644
--- a/tests/test_models.py
+++ b/tests/test_models.py
@@ -15,9 +15,9 @@ torch.manual_seed(SEED)
 np.random.seed(SEED)
 
 
-# -----------------
+# ------------------------------------------------------------------------
 # TESTS
-# -----------------
+# ------------------------------------------------------------------------
 def test_cpu_model():
     """
     Make sure model trains on CPU
@@ -222,9 +222,9 @@ def test_amp_gpu_ddp():
     clear_save_dir()
 
 
-# -----------------
+# ------------------------------------------------------------------------
 # UTILS
-# -----------------
+# ------------------------------------------------------------------------
 def get_model():
     # set up model with these hyperparams
     root_dir = os.path.dirname(os.path.realpath(__file__))