diff --git a/flake.nix b/flake.nix
index e44286d..64f69e3 100644
--- a/flake.nix
+++ b/flake.nix
@@ -83,6 +83,9 @@
             numpy
             matplotlib
           ]);
+        shellHook = ''
+          export PYTHONPATH=`pwd`/pytec:$PYTHONPATH
+        '';
       };
 
       formatter.x86_64-linux = nixpkgs.legacyPackages.x86_64-linux.nixfmt-rfc-style;