diff --git a/mach_nix/nix/nixpkgs-json.nix b/mach_nix/nix/nixpkgs-json.nix
index 75e9ebe..d3ddc8e 100644
--- a/mach_nix/nix/nixpkgs-json.nix
+++ b/mach_nix/nix/nixpkgs-json.nix
@@ -22,6 +22,10 @@ let
       });
   };
 
+  nameMap = {
+    pytorch = "torch";
+  };
+
   py = python.override { packageOverrides = mergeOverrides ( overrides ++ [ fetchPypiPnamePassthruOverride ] ); };
 in
 
@@ -46,7 +50,8 @@ let
         if hasAttrByPath ["${attrname}" "src" "pname"] python.pkgs then
           python.pkgs."${attrname}".src.pname
         else if hasAttrByPath ["${attrname}" "pname"] python.pkgs then
-          python.pkgs."${attrname}".pname
+          let pname = python.pkgs."${attrname}".pname; in
+            if nameMap ? "${pname}" then nameMap."${pname}" else pname
           else ""
       );
     in