diff --git a/flake.nix b/flake.nix
index f58279b..9dc62c4 100644
--- a/flake.nix
+++ b/flake.nix
@@ -14,11 +14,14 @@
       overlays.default = nixpkgs.lib.composeManyExtensions [
         poetry2nix.overlay
         (final: prev: {
+          inventoryDevEnv = prev.poetry2nix.mkPoetryEnv
+            {
+              projectDir = ./.;
+              groups = [ "main" "dev" ];
+            };
           inventoryEnv = prev.poetry2nix.mkPoetryEnv {
             projectDir = ./.;
-          };
-          inventoryPackage = prev.poetry2nix.mkPoetryApplication {
-            projectDir = ./.;
+            groups = [ "main" ];
           };
         })
       ];
@@ -72,7 +75,7 @@
         devShells.default = pkgs.mkShell {
           buildInputs = [
             pkgs.gnumake
-            pkgs.inventoryEnv
+            pkgs.inventoryDevEnv
             pkgs.poetry
             pkgs.python310Packages.pip
           ];
@@ -89,7 +92,7 @@
             doCheck = true;
             name = "lint";
             src = ./.;
-            checkInputs = [ pkgs.inventoryEnv ];
+            checkInputs = [ pkgs.inventoryDevEnv ];
             checkPhase = ''
               mkdir -p $out
               flake8 . --count --show-source --statistics
@@ -103,7 +106,7 @@
             doCheck = true;
             name = "test";
             src = ./.;
-            checkInputs = [ pkgs.inventoryEnv ];
+            checkInputs = [ pkgs.inventoryDevEnv ];
             checkPhase = ''
               mkdir -p $out
               pytest --ds=network_inventory.settings.ram_test \
diff --git a/pyproject.toml b/pyproject.toml
index 689226f..9783787 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -11,7 +11,7 @@ packages = [
   { include = "src" },
 ]
 
-[tool.poetry.dependencies]
+[tool.poetry.group.main.dependencies]
 python = "^3.9"
 Django = "^4.1.3"
 django-crispy-forms = "^1.14.0"
@@ -26,7 +26,7 @@ gunicorn = "^20.1.0"
 psycopg2-binary = "^2.9.5"
 PyYAML = "^6.0"
 
-[tool.poetry.dev-dependencies]
+[tool.poetry.group.dev.dependencies]
 autopep8 = "^2.0.0"
 black = "^22.10.0"
 coverage = "^6.5.0"