diff --git a/src/flake/packages/pi-coding-agent.nix b/src/flake/packages/pi-coding-agent.nix
index ce2b5df..6b46764 100644
--- a/src/flake/packages/pi-coding-agent.nix
+++ b/src/flake/packages/pi-coding-agent.nix
@@ -1,8 +1,4 @@
-{
-  pi-coding-agent,
-  root,
-}:
-pi-coding-agent.overrideAttrs {
-  src = root + "/third_party/pi";
-  patches = [ ];
+{ pkgs, root }:
+pkgs.callPackage (root + "/third_party/pi/nix/package.nix") {
+  source = root + "/third_party/pi";
 }