diff --git a/dashboard/Sugartrail.ipynb b/dashboard/Sugartrail.ipynb
index ea3b8e2..f0b5ee0 100644
--- a/dashboard/Sugartrail.ipynb
+++ b/dashboard/Sugartrail.ipynb
@@ -125,8 +125,6 @@
     "generate_network_button.on_click(lambda bt: generate_network()) \n",
     "\n",
     "def init_network():\n",
-    "    init_button.disabled=True\n",
-    "    company_text.disabled=True\n",
     "    api.basic_auth.username = API_input.value\n",
     "    response = api.get_company(str(company_text.value))\n",
     "    if response:\n",
@@ -134,11 +132,11 @@
     "        init_status.value = u'\\u2705: Initialisation successful for ' + str(response['company_name']) \n",
     "        depth_selector.disabled = False\n",
     "        generate_network_button.disabled = False\n",
+    "        init_button.disabled=True\n",
+    "        company_text.disabled=True\n",
     "    else:\n",
     "        init_button.disabled=False\n",
     "        company_text.disabled=False\n",
-    "#         auth_button.disabled=False\n",
-    "#         API_input.disabled=False\n",
     "        init_status.value = u'\\u274c: Initialisation Failed. No records for company: ' + str(company_text.value) + ' found.'\n",
     "\n",
     "display(company_text, init_button, init_status)"