diff --git a/builder/default.nix b/builder/default.nix
index 4ef5a1a..f9b14a5 100644
--- a/builder/default.nix
+++ b/builder/default.nix
@@ -285,12 +285,22 @@ let
           exclude+='\)'
 
           buildGoDir() {
-            local d; local cmd;
-            cmd="$1"
-            d="$2"
+            local cmd="$1" dir="$2"
+
             . $TMPDIR/buildFlagsArray
+
+            declare -a flags
+            flags+=($buildFlags "''${buildFlagsArray[@]}")
+            flags+=(''${tags:+-tags=${lib.concatStringsSep "," tags}})
+            flags+=(''${ldflags:+-ldflags="$ldflags"})
+            flags+=("-v" "-p" "$NIX_BUILD_CORES")
+
+            if [ "$cmd" = "test" ]; then
+              flags+=($checkFlags)
+            fi
+
             local OUT
-            if ! OUT="$(go $cmd $buildFlags "''${buildFlagsArray[@]}" ''${tags:+-tags=${concatStringsSep "," tags}} ''${ldflags:+-ldflags="$ldflags"} -v -p $NIX_BUILD_CORES $d 2>&1)"; then
+            if ! OUT="$(go $cmd "''${flags[@]}" $dir 2>&1)"; then
               if echo "$OUT" | grep -qE 'imports .*?: no Go files in'; then
                 echo "$OUT" >&2
                 return 1
@@ -352,7 +362,7 @@ let
           runHook preCheck
 
           for pkg in $(getGoDirs test); do
-            buildGoDir test $checkFlags "$pkg"
+            buildGoDir test "$pkg"
           done
 
           runHook postCheck