diff --git a/pandasai/safe_libs/restricted_matplotlib.py b/pandasai/safe_libs/restricted_matplotlib.py
index d2051975b..82635bfda 100644
--- a/pandasai/safe_libs/restricted_matplotlib.py
+++ b/pandasai/safe_libs/restricted_matplotlib.py
@@ -1,6 +1,7 @@
-import matplotlib.pyplot as plt
-import matplotlib.figure as figure
 import matplotlib.axes as axes
+import matplotlib.figure as figure
+import matplotlib.pyplot as plt
+
 from .base_restricted_module import BaseRestrictedModule