diff --git a/pkgs/by-name/am/amp/package.nix b/pkgs/by-name/am/amp/package.nix
index 2d0890b63d67..cd1e8eb7ca32 100644
--- a/pkgs/by-name/am/amp/package.nix
+++ b/pkgs/by-name/am/amp/package.nix
@@ -11,6 +11,7 @@
   libgit2,
   darwin,
   curl,
+  writableTmpDirAsHomeHook,
 }:
 
 rustPlatform.buildRustPackage rec {
@@ -20,7 +21,7 @@ rustPlatform.buildRustPackage rec {
   src = fetchFromGitHub {
     owner = "jmacdonald";
     repo = "amp";
-    rev = version;
+    tag = version;
     hash = "sha256-xNadwz2agPbxvgUqrUf1+KsWTmeNh8hJIWcNwTzzM/M=";
   };
 
@@ -48,8 +49,9 @@ rustPlatform.buildRustPackage rec {
       ]
     );
 
-  # Tests need to write to the theme directory in HOME.
-  preCheck = "export HOME=`mktemp -d`";
+  nativeCheckInputs = [
+    writableTmpDirAsHomeHook
+  ];
 
   meta = {
     description = "Modern text editor inspired by Vim";