diff --git a/default.nix b/default.nix
index 23410c3..f1de3e7 100644
--- a/default.nix
+++ b/default.nix
@@ -8,7 +8,8 @@
         (import "${fetchTree gomod2nix.locked}/overlay.nix")
       ];
     }
-  )
+  ),
+  lib
 }:
 
 pkgs.buildGoApplication {
@@ -18,4 +19,13 @@ pkgs.buildGoApplication {
   src = ./.;
   modules = ./gomod2nix.toml;
   doCheck = false;
+
+  buildInputs = [ pkgs.makeWrapper ];
+
+  postBuild = ''
+    wrapProgram $out/bin/p2p-rendering-computation \
+      --set PATH  $out/bin \
+      --set P2PRC $out/bin
+  '';
+
 }
diff --git a/flake.nix b/flake.nix
index 15e7328..068e4a5 100644
--- a/flake.nix
+++ b/flake.nix
@@ -23,11 +23,8 @@
           callPackage = pkgs.darwin.apple_sdk_11_0.callPackage or pkgs.callPackage;
         in
         {
-          # packages.default = callPackage ./. {
-            # inherit (gomod2nix.legacyPackages.${system}) buildGoApplication;
-          # };
 
-          packages.default = pkgs.callPackage ./. { };
+          packages.default = callPackage ./. { };
 
           devShells.default = pkgs.mkShell {
             buildInputs = with pkgs; [