diff --git a/flake.nix b/flake.nix
index ff9dc15..00e178a 100644
--- a/flake.nix
+++ b/flake.nix
@@ -8,7 +8,12 @@
   outputs = { self, nixpkgs, flake-utils }:
     flake-utils.lib.eachDefaultSystem (system:
       let
-        pkgs = nixpkgs.legacyPackages.${system};
+        pkgs = import nixpkgs {
+          system = system;
+          config.allowUnfree = true;
+          config.cudaSupport = true;
+        };
+        # pkgs = nixpkgs.legacyPackages.${system};
 
         pythonPackages = pkgs.python312Packages;
 
@@ -189,8 +194,7 @@
           python312Packages.opencv4 = custom-opencv4;
         };
         devShells.default = pkgs.mkShell {
-          buildInputs =
-            [ (pkgs.python312.withPackages (ps: [ custom-opencv4 ])) ];
+          buildInputs = [ (pkgs.python312.withPackages (ps: [ ])) ];
         };
       });
 }