diff --git a/examples/camels.ipynb b/examples/camels.ipynb
index e1cdd1e..1df49ed 100644
--- a/examples/camels.ipynb
+++ b/examples/camels.ipynb
@@ -17,7 +17,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from seshat_api import SeshatAPI, get_variable_classes\n",
+    "from seshat_api import SeshatAPI\n",
     "import pandas as pd\n",
     "import matplotlib.pyplot as plt"
    ]
diff --git a/tests/test_seshat_api.py b/tests/test_seshat_api.py
index ce85ba8..332717c 100644
--- a/tests/test_seshat_api.py
+++ b/tests/test_seshat_api.py
@@ -27,6 +27,7 @@ def test_get_variable_classes():
     assert "BigPonies" not in vc
 
 
+# TODO: Use mocking instead of relying on the API
 def test_get_frequencies():
     # Only run this test if we can access the API url
     base_url = "https://seshatdata.com/api"