diff --git a/pkgs/by-name/nv/nvitop/package.nix b/pkgs/by-name/nv/nvitop/package.nix
index c52d481caf02..6c7bb3108118 100644
--- a/pkgs/by-name/nv/nvitop/package.nix
+++ b/pkgs/by-name/nv/nvitop/package.nix
@@ -2,38 +2,40 @@
   lib,
   python3Packages,
   fetchFromGitHub,
+  versionCheckHook,
 }:
 
 python3Packages.buildPythonApplication rec {
   pname = "nvitop";
-  version = "1.3.2";
+  version = "1.4.0";
 
   src = fetchFromGitHub {
     owner = "XuehaiPan";
-    repo = pname;
-    rev = "refs/tags/v${version}";
-    hash = "sha256-TunGtNe+lgx/hk8kNtB8yaCdbkiJ3d4JJ8NKB+6urJA=";
+    repo = "nvitop";
+    tag = "v${version}";
+    hash = "sha256-Z0JGW7vcZFxguQqhqhpPpOvcOct7B9z8RoEFu5NsOl0=";
   };
 
   pythonRelaxDeps = [ "nvidia-ml-py" ];
 
-  propagatedBuildInputs = with python3Packages; [
+  dependencies = with python3Packages; [
     cachetools
     psutil
     termcolor
     nvidia-ml-py
   ];
 
-  checkPhase = ''
-    $out/bin/nvitop --help
-  '';
+  nativeCheckInputs = [
+    versionCheckHook
+  ];
+  versionCheckProgramArg = [ "--version" ];
 
-  meta = with lib; {
+  meta = {
     description = "Interactive NVIDIA-GPU process viewer, the one-stop solution for GPU process management";
     homepage = "https://github.com/XuehaiPan/nvitop";
     changelog = "https://github.com/XuehaiPan/nvitop/releases/tag/v${version}";
-    license = licenses.gpl3;
-    maintainers = with maintainers; [ GaetanLepage ];
-    platforms = with platforms; linux;
+    license = lib.licenses.gpl3;
+    maintainers = with lib.maintainers; [ GaetanLepage ];
+    platforms = with lib.platforms; linux;
   };
 }