diff --git a/pkgs/development/python-modules/torchtitan/default.nix b/pkgs/development/python-modules/torchtitan/default.nix
index e7e16da4040f..da624b457808 100644
--- a/pkgs/development/python-modules/torchtitan/default.nix
+++ b/pkgs/development/python-modules/torchtitan/default.nix
@@ -20,6 +20,7 @@
   tyro,
 
   # tests
+  expecttest,
   pytestCheckHook,
   tomli-w,
   triton,
@@ -58,6 +59,7 @@ buildPythonPackage (finalAttrs: {
   pythonImportsCheck = [ "torchtitan" ];
 
   nativeCheckInputs = [
+    expecttest
     pytestCheckHook
     tomli-w
     transformers