diff --git a/frontend/pages/prediction_classification.py b/frontend/pages/prediction_classification.py
index bb6bb22..648db06 100644
--- a/frontend/pages/prediction_classification.py
+++ b/frontend/pages/prediction_classification.py
@@ -74,6 +74,6 @@ if "data" in st.session_state:
             plt.xlabel('Predicted')
             plt.ylabel('True')
 
-            st.pyplot(fig, figsize=(1, 1))
+            st.pyplot(fig)
 else:
     st.error("File not loaded")