diff --git a/src/HowBadIsMyBatch.ipynb b/src/HowBadIsMyBatch.ipynb
index ed858432ac7..5d75ae89f97 100644
--- a/src/HowBadIsMyBatch.ipynb
+++ b/src/HowBadIsMyBatch.ipynb
@@ -261,13 +261,10 @@
     "\n",
     "    @staticmethod\n",
     "    def createCompanyByBatchCodeTable(dataFrame):\n",
-    "        return CompanyColumnAdder._createManufacturerByBatchCodeTable(dataFrame).rename(columns = {\"VAX_MANU\": \"Company\"})\n",
-    "\n",
-    "    @staticmethod\n",
-    "    def _createManufacturerByBatchCodeTable(dataFrame):\n",
     "        manufacturerByBatchCodeTable = dataFrame[['VAX_LOT', 'VAX_MANU']]\n",
     "        manufacturerByBatchCodeTable = manufacturerByBatchCodeTable.drop_duplicates(subset = ['VAX_LOT'])\n",
-    "        return manufacturerByBatchCodeTable.set_index('VAX_LOT')\n"
+    "        manufacturerByBatchCodeTable = manufacturerByBatchCodeTable.set_index('VAX_LOT')\n",
+    "        return manufacturerByBatchCodeTable.rename(columns = {\"VAX_MANU\": \"Company\"})"
    ]
   },
   {
@@ -730,6 +727,14 @@
     "    country = 'Global',\n",
     "    minADRsForLethality = minADRsForLethality)"
    ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "258adf58",
+   "metadata": {},
+   "outputs": [],
+   "source": []
   }
  ],
  "metadata": {