diff --git a/nixos/modules/hardware/tenstorrent.nix b/nixos/modules/hardware/tenstorrent.nix
index 30840021b714..c042c600a86b 100644
--- a/nixos/modules/hardware/tenstorrent.nix
+++ b/nixos/modules/hardware/tenstorrent.nix
@@ -24,7 +24,7 @@ in
     ];
 
     environment.systemPackages = with pkgs; [
-      python3Packages.tt-smi
+      tt-smi
       tt-system-tools
     ];
   };