diff --git a/pkgs/development/rocm-modules/6/amdsmi/default.nix b/pkgs/development/rocm-modules/6/amdsmi/default.nix
index 6f26889ab01b..a145e4d61ded 100644
--- a/pkgs/development/rocm-modules/6/amdsmi/default.nix
+++ b/pkgs/development/rocm-modules/6/amdsmi/default.nix
@@ -70,6 +70,7 @@ stdenv.mkDerivation (finalAttrs: {
   ];
 
   postInstall = ''
+    makeWrapperArgs=(--prefix LD_LIBRARY_PATH : ${lib.makeLibraryPath [ libdrm ]})
     wrapPythonProgramsIn $out
     rm $out/bin/amd-smi
     ln -sf $out/libexec/amdsmi_cli/amdsmi_cli.py $out/bin/amd-smi
@@ -88,5 +89,6 @@ stdenv.mkDerivation (finalAttrs: {
     maintainers = with maintainers; [ lovesegfault ];
     teams = [ teams.rocm ];
     platforms = [ "x86_64-linux" ];
+    mainProgram = "amd-smi";
   };
 })