diff --git a/tests/python_package_test/test_plotting.py b/tests/python_package_test/test_plotting.py
index 2c25d5733ed0..6ffb7a1602d6 100644
--- a/tests/python_package_test/test_plotting.py
+++ b/tests/python_package_test/test_plotting.py
@@ -1,8 +1,9 @@
 # coding: utf-8
-import lightgbm as lgb
-from lightgbm.compat import MATPLOTLIB_INSTALLED, GRAPHVIZ_INSTALLED
-from sklearn.model_selection import train_test_split
 import pytest
+from sklearn.model_selection import train_test_split
+
+import lightgbm as lgb
+from lightgbm.compat import GRAPHVIZ_INSTALLED, MATPLOTLIB_INSTALLED
 
 if MATPLOTLIB_INSTALLED:
     import matplotlib