diff --git a/dashboard/Sugartrail.ipynb b/dashboard/Sugartrail.ipynb
index ce41bda..edf5cad 100644
--- a/dashboard/Sugartrail.ipynb
+++ b/dashboard/Sugartrail.ipynb
@@ -130,8 +130,9 @@
     "    api.basic_auth.username = API_input.value\n",
     "    response = api.get_company(str(company_text.value))\n",
     "    if response:\n",
-    "        network.company_id = str(company_text.value)\n",
-    "#         init_status.value = u'\\u2705: Initialisation successful for ' + str(response['company_name']) \n",
+    "        network = base.Network(str(company_text.value))\n",
+    "#         network.company_id = str(company_text.value)\n",
+    "        init_status.value = u'\\u2705: Initialisation successful for ' + str(response['company_name']) \n",
     "        depth_selector.disabled = False\n",
     "        generate_network_button.disabled = False\n",
     "    else:\n",