diff --git a/project/ui/qprof_main.ipynb b/project/ui/qprof_main.ipynb
index 50923a00..d1c7e613 100644
--- a/project/ui/qprof_main.ipynb
+++ b/project/ui/qprof_main.ipynb
@@ -950,15 +950,15 @@
     "target_schema_2_combo = widgets.HBox([target_schema_2,\n",
     "                                      create_tooltip(\"Enter the Target Schema of the saved query profile. If you enter this, the below transaction id and statement id will not be used.\"),\n",
     "                                      widgets.HTML(html_arrow),schema_dropdown_comp2])\n",
-    "key_2 = widgets.Text(description=\"Key ID\", placeholder=\"Key ID for Query # 2\", layout=widgets.Layout(width='300px'))\n",
-    "key_2.style.description_width = '100px'\n",
-    "key_2_combo = widgets.HBox([key_2,\n",
+    "key_comp_2 = widgets.Text(description=\"Key ID\", placeholder=\"Key ID for Query # 2\", layout=widgets.Layout(width='300px'))\n",
+    "key_comp_2.style.description_width = '100px'\n",
+    "key_comp_2_combo = widgets.HBox([key_comp_2,\n",
     "                            create_tooltip(\"Enter the unique Key ID. If you enter this, the below transaction id and statement id will not be used\"),\n",
     "                            widgets.HTML(html_arrow),\n",
     "                            key_dropdown_comp2])\n",
     "# Set observers for schema_dropdown_comp2 and\n",
     "schema_dropdown_comp2.observe(update_key_dropdown_comp2_options, names='value')\n",
-    "key_2.observe(observe_value_change_comp2, names='value')\n",
+    "key_comp_2.observe(observe_value_change_comp2, names='value')\n",
     "target_schema_2.observe(observe_value_change_comp2, names='value')\n",
     "\n",
     "transaction_id_2 = widgets.Text(description=\"Transaction ID\", placeholder=\"Transaction ID for Query # 2\", layout=widgets.Layout(width='300px'))\n",
@@ -979,7 +979,7 @@
     "session_control_2.style.description_width = '150px'\n",
     "session_control_2_combo = widgets.HBox([session_control_2, create_tooltip(\"Enter the session control SQL that you want to run before your query. This is useful if you want to control the session parameters for your query. For example, you can use this to set the statement memory for your query. You can even enter multiple statements by separating them with a semicolon.\"),])\n",
     "\n",
-    "comparison_section_query_2 = widgets.VBox([comparison_section_heading_query_2, target_schema_2_combo, key_2_combo, transaction_id_2_combo, statement_id_2_combo, query_2_combo, session_control_2_combo])\n",
+    "comparison_section_query_2 = widgets.VBox([comparison_section_heading_query_2, target_schema_2_combo, key_comp_2_combo, transaction_id_2_combo, statement_id_2_combo, query_2_combo, session_control_2_combo])\n",
     "comparison_section_query_2.layout.width = \"50%\"\n",
     "# Combine both query sections in a HBox\n",
     "comparison_section_boxes = widgets.HBox([comparison_section_query_1, comparison_section_query_2])\n",
@@ -1018,7 +1018,7 @@
     "        transaction_comp2_val = transaction_id_2.value\n",
     "        statement_comp2_val = statement_id_2.value  # Capture Statement ID\n",
     "        target_schema_comp2_val = target_schema_2.value if target_schema_2.value != \"\" else schema_dropdown_comp2.value # Capture Target Schema\n",
-    "        key_comp2_val = key_2.value if key_2.value != \"\" else key_dropdown_comp2.value # Capture Key ID\n",
+    "        key_comp2_val = key_comp_2.value if key_comp_2.value != \"\" else key_dropdown_comp2.value # Capture Key ID\n",
     "        query_comp2_val = query_2.value  # Capture SQL Query\n",
     "        session_control_2_val = session_control_2.value\n",
     "        \n",