diff --git a/tests/notebooks/test_run_notebooks.py b/tests/notebooks/test_run_notebooks.py
index 4ca85405..db45f01d 100644
--- a/tests/notebooks/test_run_notebooks.py
+++ b/tests/notebooks/test_run_notebooks.py
@@ -12,8 +12,6 @@
     torch_geometric_available,
 )
 
-# TODO @cog-imperial: These will be replaced with stronger tests using testbook soon
-# https://github.com/cog-imperial/OMLT/issues/159
 
 def _test_run_notebook(folder, notebook_fname, n_cells):
     # Change to notebook directory to allow for testing