diff --git a/nixos/modules/image/amend-repart-definitions.py b/nixos/modules/image/amend-repart-definitions.py
index 939340266ab7..be8a207726fd 100644
--- a/nixos/modules/image/amend-repart-definitions.py
+++ b/nixos/modules/image/amend-repart-definitions.py
@@ -13,8 +13,8 @@ files using the same mechanism.
 """
 
 import json
-import sys
 import shutil
+import sys
 from pathlib import Path
 
 
diff --git a/nixos/modules/image/repart-image.nix b/nixos/modules/image/repart-image.nix
index 9b4447b18816..129aad48d2f8 100644
--- a/nixos/modules/image/repart-image.nix
+++ b/nixos/modules/image/repart-image.nix
@@ -86,7 +86,6 @@ let
         depsBuildBuild = [ ruff ];
         nativeBuildInputs = [
           python3
-          black
           mypy
         ];
       }
@@ -94,8 +93,8 @@ let
         install ${./amend-repart-definitions.py} $out
         patchShebangs --build $out
 
-        black --check --diff $out
-        ruff check --line-length 88 $out
+        ruff format --check $out
+        ruff check $out
         mypy --strict $out
       '';