diff --git a/pkgs/development/python-modules/langgraph-prebuilt/default.nix b/pkgs/development/python-modules/langgraph-prebuilt/default.nix
index 73a1d6ade3ad..995c7f552494 100644
--- a/pkgs/development/python-modules/langgraph-prebuilt/default.nix
+++ b/pkgs/development/python-modules/langgraph-prebuilt/default.nix
@@ -16,11 +16,12 @@
   langgraph-checkpoint-sqlite,
   postgresql,
   postgresqlTestHook,
-  psycopg-pool,
   psycopg,
+  psycopg-pool,
   pytest-asyncio,
   pytest-mock,
   pytestCheckHook,
+  xxhash,
 
   # passthru
   nix-update-script,
@@ -55,19 +56,17 @@ buildPythonPackage rec {
   doCheck = !stdenv.hostPlatform.isDarwin;
 
   nativeCheckInputs = [
-    pytestCheckHook
-    postgresql
-    postgresqlTestHook
-  ];
-
-  checkInputs = [
     langgraph-checkpoint
     langgraph-checkpoint-postgres
     langgraph-checkpoint-sqlite
+    postgresql
+    postgresqlTestHook
     psycopg
     psycopg-pool
     pytest-asyncio
     pytest-mock
+    pytestCheckHook
+    xxhash
   ];
 
   preCheck = ''
diff --git a/pkgs/development/python-modules/langgraph/default.nix b/pkgs/development/python-modules/langgraph/default.nix
index 2716824e6d71..708da84981d8 100644
--- a/pkgs/development/python-modules/langgraph/default.nix
+++ b/pkgs/development/python-modules/langgraph/default.nix
@@ -23,7 +23,6 @@
   langgraph-checkpoint-sqlite,
   langsmith,
   psycopg,
-  psycopg-pool,
   pydantic,
   pytest-asyncio,
   pytest-mock,
@@ -115,10 +114,11 @@ buildPythonPackage rec {
 
   disabledTestPaths = [
     # psycopg.errors.InsufficientPrivilege: permission denied to create database
-    "tests/test_pregel_async.py"
-    "tests/test_pregel.py"
+    "tests/test_checkpoint_migration.py"
     "tests/test_large_cases.py"
     "tests/test_large_cases_async.py"
+    "tests/test_pregel.py"
+    "tests/test_pregel_async.py"
   ];
 
   passthru = {