diff --git a/test.py b/test.py
index 3b0e18b..0416a70 100644
--- a/test.py
+++ b/test.py
@@ -22,6 +22,7 @@ model_bytes = open("models/mobilenet_v3.keras", "rb")
 # Convert model to Keras model
 model_file = open("model.keras", "wb").write(model_bytes.read())
 model = tf.keras.models.load_model("model.keras")
+os.remove("model.keras")
 
 
 benign_image = preprocess_image_bytes(benign_image_bytes)
@@ -30,7 +31,6 @@ malignant_image = preprocess_image_bytes(malignant_image_bytes)
 benign_prediction = model.predict(benign_image)
 malignant_prediction = model.predict(malignant_image)
 
-os.remove("model.keras")
 
 print(f"Benign Prediction: {benign_prediction}")
 print(f"Malignant Prediction: {malignant_prediction}")