1 files changed, 2 insertions, 3 deletions

diff --git a/ymir.nix b/ymir.nix
index 7a514c01..0355540d 100644
--- a/ymir.nix
+++ b/ymir.nix
@@ -62,12 +62,11 @@ in rec {
         buildInputs = oldAttrs.buildInputs ++ [
           pkgs.perl
           pkgs.python2
-          pkgs.python2Packages.wrapPython
           pkgs.makeWrapper
         ];
-        pythonPath = with pkgs.python2Packages ; [pygments];
         postInstall = ''
-          wrapPythonPrograms
+          wrapProgram $out/lib/cgit/filters/syntax-highlighting.py \
+            --prefix PYTHONPATH ':' ${pkgs.python.buildEnv.override { extraLibs = [ pygments ]; }}/lib/*/site-packages
         '';
       });
     };