diff --git a/flake.nix b/flake.nix
index 8c81be8..5cba347 100644
--- a/flake.nix
+++ b/flake.nix
@@ -95,6 +95,7 @@
           ++ (with python3Packages; [
             numpy
             matplotlib
+            pythermostat
           ]);
         shellHook = ''
           export PYTHONPATH=`git rev-parse --show-toplevel`/pythermostat:$PYTHONPATH