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
     ];
   };
diff --git a/pkgs/development/python-modules/tt-smi/default.nix b/pkgs/by-name/tt/tt-smi/package.nix
similarity index 73%
rename from pkgs/development/python-modules/tt-smi/default.nix
rename to pkgs/by-name/tt/tt-smi/package.nix
index c58ae346e466..fad569cf70db 100644
--- a/pkgs/development/python-modules/tt-smi/default.nix
+++ b/pkgs/by-name/tt/tt-smi/package.nix
@@ -1,21 +1,11 @@
 {
   lib,
-  buildPythonPackage,
+  python3Packages,
   fetchFromGitHub,
-  pythonOlder,
-  setuptools,
-  distro,
-  elasticsearch,
-  pydantic,
-  pyluwen,
-  rich,
-  textual,
   pre-commit,
-  importlib-resources,
-  tt-tools-common,
-  tomli,
+  versionCheckHook,
 }:
-buildPythonPackage rec {
+python3Packages.buildPythonApplication rec {
   pname = "tt-smi";
   version = "3.0.30";
   pyproject = true;
@@ -27,13 +17,11 @@ buildPythonPackage rec {
     hash = "sha256-C6CfcS0H3rFew/Y1uhmzICdFp1UYU7H9h3YPeAKlcbE=";
   };
 
-  disabled = pythonOlder "3.13";
-
-  build-system = [
+  build-system = with python3Packages; [
     setuptools
   ];
 
-  dependencies = [
+  dependencies = with python3Packages; [
     distro
     elasticsearch
     pydantic
@@ -47,10 +35,17 @@ buildPythonPackage rec {
     tomli
   ];
 
+  nativeCheckInputs = [
+    versionCheckHook
+  ];
+
   # Fails due to having no tests
   dontUsePytestCheck = true;
 
+  versionCheckProgramArg = "--version";
+
   meta = {
+    mainProgram = "tt-smi";
     description = "Tenstorrent console based hardware information program";
     homepage = "https://github.com/tenstorrent/tt-smi";
     maintainers = with lib.maintainers; [ RossComputerGuy ];
diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix
index 1a61be6e913c..beb96bbe5f4f 100644
--- a/pkgs/top-level/python-packages.nix
+++ b/pkgs/top-level/python-packages.nix
@@ -18855,8 +18855,6 @@ self: super: with self; {
 
   tt-flash = callPackage ../development/python-modules/tt-flash { };
 
-  tt-smi = callPackage ../development/python-modules/tt-smi { };
-
   tt-tools-common = callPackage ../development/python-modules/tt-tools-common { };
 
   ttach = callPackage ../development/python-modules/ttach { };