diff --git a/paper2_LIME/Solubility-RNN.ipynb b/paper2_LIME/Solubility-RNN.ipynb
index e37352f..79a9999 100644
--- a/paper2_LIME/Solubility-RNN.ipynb
+++ b/paper2_LIME/Solubility-RNN.ipynb
@@ -302,8 +302,8 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "model.save(\"solubility-rnn-accurate\")\n",
-    "# model = tf.keras.models.load_model('solubility-rnn-accurate/')"
+    "model.save(\"solubility-rnn-accurate.keras\")\n",
+    "# model = tf.keras.models.load_model('solubility-rnn-accurate.keras')"
    ]
   },
   {