diff --git a/curriculum/challenges/english/07-scientific-computing-with-python/learn-tree-traversal-by-building-a-binary-search-tree/65c4eb814cc977c95cd7df0e.md b/curriculum/challenges/english/07-scientific-computing-with-python/learn-tree-traversal-by-building-a-binary-search-tree/65c4eb814cc977c95cd7df0e.md
index c18286d138c..66aa0bf873d 100644
--- a/curriculum/challenges/english/07-scientific-computing-with-python/learn-tree-traversal-by-building-a-binary-search-tree/65c4eb814cc977c95cd7df0e.md
+++ b/curriculum/challenges/english/07-scientific-computing-with-python/learn-tree-traversal-by-building-a-binary-search-tree/65c4eb814cc977c95cd7df0e.md
@@ -24,8 +24,7 @@ You should use the `class` keyword to declare an empty class named `TreeNode`.
   test: () => {
     assert(__pyodide.runPython(`
       import inspect
-      a = __locals.get("TreeNode")
-      inspect.isclass(a)
+      inspect.isclass(TreeNode)
     `));
   }
 })
diff --git a/curriculum/challenges/english/07-scientific-computing-with-python/learn-tree-traversal-by-building-a-binary-search-tree/65c4f09e074dd8d37830ea00.md b/curriculum/challenges/english/07-scientific-computing-with-python/learn-tree-traversal-by-building-a-binary-search-tree/65c4f09e074dd8d37830ea00.md
index bcaae8c345f..4cc5678f64a 100644
--- a/curriculum/challenges/english/07-scientific-computing-with-python/learn-tree-traversal-by-building-a-binary-search-tree/65c4f09e074dd8d37830ea00.md
+++ b/curriculum/challenges/english/07-scientific-computing-with-python/learn-tree-traversal-by-building-a-binary-search-tree/65c4f09e074dd8d37830ea00.md
@@ -18,8 +18,7 @@ You should have a class named `BinarySearchTree`. Remember to use the `pass` key
   test: () => {
     assert(__pyodide.runPython(`
       import inspect
-      bst = __locals.get("BinarySearchTree")
-      inspect.isclass(bst)
+      inspect.isclass(BinarySearchTree)
     `));
   }
 })