diff --git a/pyproject.toml b/pyproject.toml
index 2da9eba..36add74 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -43,7 +43,7 @@ extra-dependencies = [
   "pytest",
   "llama-index-core",
   "smolagents",
-  "langchain",
+  "langchain-core",
 ]
 [tool.hatch.envs.types.scripts]
 check = "mypy --strict --install-types --non-interactive {args:src/cleanlab_codex tests}"
@@ -54,7 +54,7 @@ allow-direct-references = true
 extra-dependencies = [
   "llama-index-core",
   "smolagents; python_version >= '3.10'",
-  "langchain_core",
+  "langchain-core",
 ]
 
 [tool.hatch.envs.hatch-test.env-vars]