diff --git a/packages/edk2/default.nix b/packages/edk2/default.nix
index f4a2878..b51f739 100644
--- a/packages/edk2/default.nix
+++ b/packages/edk2/default.nix
@@ -67,6 +67,8 @@ stdenv.mkDerivation rec {
 
     chmod -R u+w .
 
+    patchShebangs ./edk2
+
     runHook postUnpack
   '';
 
@@ -85,12 +87,15 @@ stdenv.mkDerivation rec {
 
   buildPhase = ''
     export PYTHON_COMMAND=${python3}/bin/python
+    export WORKSPACE=$PWD
+    export PACKAGES_PATH=$WORKSPACE/edk2:$WORKSPACE/edk2-non-osi:$WORKSPACE/edk2-platforms
+    export EDK_TOOLS_PATH=$WORKSPACE/edk2/BaseTools
 
     make -C ./edk2/BaseTools
 
     export ATF_BUILD_DIR="${pkgs.${namespace}.arm-trusted-firmware}/rpi${MODEL}/${RELEASE_TYPE}"
 
-    source ./edk2/edksetup.sh
+    . ./edk2/edksetup.sh BaseTools
 
     build \
         -a AARCH64 \
@@ -113,4 +118,4 @@ stdenv.mkDerivation rec {
     platforms = [ "aarch64-linux" ];
     maintainers = with maintainers; [ kazenyuk ];
   };
-}
\ No newline at end of file
+}