diff --git a/flake.nix b/flake.nix
index c846d21..f342daf 100644
--- a/flake.nix
+++ b/flake.nix
@@ -95,6 +95,9 @@
             numpy
             matplotlib
           ]);
+        shellHook = ''
+          export PYTHONPATH=`git rev-parse --show-toplevel`/pythermostat:$PYTHONPATH
+        '';
       };
 
       formatter.x86_64-linux = nixpkgs.legacyPackages.x86_64-linux.nixfmt-rfc-style;