diff --git a/templates/python/.envrc b/templates/python/.envrc
index 8c7c1c9..8fa86e1 100644
--- a/templates/python/.envrc
+++ b/templates/python/.envrc
@@ -1,4 +1,4 @@
-use flake
+use nix
 export PYTHONPATH="$(pwd)/src:$PYTHONPATH"
 if [ ! -z $NIX_LD ]; then
   export LD_LIBRARY_PATH=$NIX_LD_LIBRARY_PATH
diff --git a/templates/python/flake.nix b/templates/python/flake.nix
deleted file mode 100644
index 8a0943a..0000000
--- a/templates/python/flake.nix
+++ /dev/null
@@ -1,15 +0,0 @@
-{
-  description = "Simple python flake";
-
-  inputs.nixpkgs.url = "github:NixOS/nixpkgs/nixos-unstable";
-
-  outputs = inputs@{ flake-parts, ... }:
-    flake-parts.lib.mkFlake { inherit inputs; } {
-      systems = [ "x86_64-linux" "aarch64-linux" "aarch64-darwin" "x86_64-darwin" ];
-      perSystem = { config, self', inputs', pkgs, system, ... }: {
-        devShells.default = pkgs.mkShell {
-          buildInputs = with pkgs; [ python3 poetry ];
-        };
-      };
-    };
-}
diff --git a/templates/python/shell.nix b/templates/python/shell.nix
new file mode 100644
index 0000000..6196009
--- /dev/null
+++ b/templates/python/shell.nix
@@ -0,0 +1,5 @@
+{ pkgs ? import <nixpkgs> { } }:
+
+pkgs.mkShell {
+  buildInputs = with pkgs; [ python3 poetry ];
+}