diff --git a/tests/test_models.py b/tests/test_models.py
index 1717adc5df..4bcae38064 100644
--- a/tests/test_models.py
+++ b/tests/test_models.py
@@ -35,9 +35,10 @@ def test_cpu_model():
 
     run_gpu_model_test(trainer_options, on_gpu=False)
 
+
 def test_all_features_cpu_model():
     """
-    Make sure model trains on CPU
+    Test each of the trainer options
     :return:
     """