diff --git a/src/methods/build-function.ts b/src/methods/build-function.ts
index bd3417a..6079c35 100644
--- a/src/methods/build-function.ts
+++ b/src/methods/build-function.ts
@@ -47,7 +47,11 @@ export async function buildFunction(functionPath?: string): Promise<BuildFunctio
         '--path', functionName
       ], {
         cwd: appRootDir,
-        stdio: ['pipe', 'pipe', 'pipe']
+        stdio: ['pipe', 'pipe', 'pipe'],
+        env: {
+          ...process.env,
+          SHOPIFY_INVOKED_BY: 'shopify-function-test-helpers'
+        }
       });
 
       let stdout = '';
diff --git a/src/methods/get-function-info.ts b/src/methods/get-function-info.ts
index 2d15be7..a84ac74 100644
--- a/src/methods/get-function-info.ts
+++ b/src/methods/get-function-info.ts
@@ -33,7 +33,11 @@ export async function getFunctionInfo(functionDir: string): Promise<FunctionInfo
       '--path', functionName
     ], {
       cwd: appRootDir,
-      stdio: ['pipe', 'pipe', 'pipe']
+      stdio: ['pipe', 'pipe', 'pipe'],
+      env: {
+        ...process.env,
+        SHOPIFY_INVOKED_BY: 'shopify-function-test-helpers'
+      }
     });
 
     let stdout = '';