diff --git a/flake.nix b/flake.nix
index 0348feb..d21b27f 100644
--- a/flake.nix
+++ b/flake.nix
@@ -41,9 +41,6 @@
         };
 
         ultralytics = pkgs.python3Packages.buildPythonPackage rec {
-          # pname = "opencv-python";
-          # version = "4.10.0.84";
-
           version = "8.3.13";
           pname = "ultralytics";
           pyproject = true;
@@ -54,6 +51,7 @@
             hash = "sha256-Wq2pCD0l0AbdQ2IPfEfxPffSxgv/5qFadB4BHAExEbU=";
             fetchSubmodules = true;
           };
+          # doCheck = false;
 
           propagatedBuildInputs = [
             pythonPackages.requests
@@ -65,35 +63,30 @@
             pythonPackages.pandas
             pythonPackages.matplotlib
             pythonPackages.seaborn
-            ultralytics-thop
+            pythonPackages.py-cpuinfo
+            pythonPackages.opencv4
 
+            ultralytics-thop
           ];
 
+          pythonRelaxDeps = [ "torchvision" ];
+          pythonRemoveDeps = [ "opencv-python" ];
+
           nativeBuildInputs = [
             pythonPackages.setuptools
             pythonPackages.setuptools-scm
             pythonPackages.pip
-            # pythonPackages.scikit-build
-            # pythonPackages.numpy_2
-            # pythonPackages.cmake
           ];
 
-          nativeCheckInputs =
-            [ pythonPackages.more-itertools pythonPackages.pytestCheckHook ];
-
-          # disabledTests = [
-          #   # creates a file, checks when it was last accessed/modified
-          #   # AssertionError: assert 1650036414.0 == 1650036414.960688
-          #   # "test_utime"
-          # ];
-
-          # pythonImportsCheck = [ "cv2" ];
+          # nativeCheckInputs = [ ];
 
           meta = with pkgs.lib; {
-            description = "Object-oriented file system path manipulation";
-            homepage = "https://github.com/jaraco/path";
-            changelog =
-              "https://github.com/jaraco/path/blob/v${version}/NEWS.rst";
+            description = ''
+              Ultralytics YOLO for SOTA object detection,
+              multi-object tracking, instance segmentation,
+              pose estimation and image classification.
+            '';
+            homepage = "https://github.com/ultralytics/ultralytics";
             license = pkgs.lib.licenses.mit;
             maintainers = with maintainers; [ ];
           };