diff --git a/pkgs/by-name/tt/tt-smi/package.nix b/pkgs/by-name/tt/tt-smi/package.nix
index 9fced72b8bfd..912e59170ec4 100644
--- a/pkgs/by-name/tt/tt-smi/package.nix
+++ b/pkgs/by-name/tt/tt-smi/package.nix
@@ -4,17 +4,19 @@
   fetchFromGitHub,
   pre-commit,
   versionCheckHook,
+  tt-umd,
 }:
 python3Packages.buildPythonApplication (finalAttrs: {
   pname = "tt-smi";
-  version = "3.0.30";
+  version = "5.2.0";
   pyproject = true;
+  __structuredAttrs = true;
 
   src = fetchFromGitHub {
     owner = "tenstorrent";
     repo = "tt-smi";
     tag = "v${finalAttrs.version}";
-    hash = "sha256-C6CfcS0H3rFew/Y1uhmzICdFp1UYU7H9h3YPeAKlcbE=";
+    hash = "sha256-meDqvDvGBXx/zbHbtgLUb+Kv8LSmsu9OvYCFhmEPAdQ=";
   };
 
   build-system = with python3Packages; [
@@ -33,8 +35,11 @@ python3Packages.buildPythonApplication (finalAttrs: {
     tt-tools-common
     setuptools
     tomli
+    tt-umd
   ];
 
+  pythonRelaxDeps = [ "tt-umd" ];
+
   nativeCheckInputs = [
     versionCheckHook
   ];