diff --git a/SalesAnalysis/SalesAnalysis.ipynb b/SalesAnalysis/SalesAnalysis.ipynb
index 9181092..95b81b4 100644
--- a/SalesAnalysis/SalesAnalysis.ipynb
+++ b/SalesAnalysis/SalesAnalysis.ipynb
@@ -816,7 +816,7 @@
     "    return address.split(\",\")[1].strip(\" \")\n",
     "\n",
     "def get_state(address):\n",
-    "    return address.split(\",\")[2].split(\" \")[1]\n",
+    "    return address.split(',')[2][1:3]\n",
     "\n",
     "all_data['City'] = all_data['Purchase Address'].apply(lambda x: f\"{get_city(x)}  ({get_state(x)})\")\n",
     "all_data.head()"