diff --git a/pkgs/smi/default.nix b/pkgs/smi/default.nix
index b88382d..325e4b0 100644
--- a/pkgs/smi/default.nix
+++ b/pkgs/smi/default.nix
@@ -54,4 +54,10 @@ buildPythonPackage rec {
   ];
 
   dontUsePytestCheck = true; # no tests
+
+  installCheckPhase = ''
+    output=$($out/bin/tt-smi || true)
+    echo "tt-smi output: $output"
+    echo $output | grep -q "No Tenstorrent driver detected"
+  '';
 }