diff --git a/requirements.txt b/requirements.txt
index 22897d4..b39e670 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,6 +2,7 @@ langchain
 langchain-core
 langchain-ollama
 langchain-community
+langchain-experimental
 langgraph
 seaborn
 pandas
diff --git a/suite_settings/tests.py b/suite_settings/tests.py
index 2190cd8..b67c731 100644
--- a/suite_settings/tests.py
+++ b/suite_settings/tests.py
@@ -55,7 +55,7 @@ tests = {
         runnable_input={
             "system_msg": 'You are a helpful assistant.',
             "human_msg": 'Is 31515261 divisible by 425? If not, whats the remainder?',
-            "tools": { "python_repl": save_python_repl },
+            "tools": { "save_python_repl": save_python_repl },
         },
         validator=regex_match_any,
         validation_input={