diff --git a/extension/flat_tensor/test/targets.bzl b/extension/flat_tensor/test/targets.bzl
index 813a1b56d98..aba9db22834 100644
--- a/extension/flat_tensor/test/targets.bzl
+++ b/extension/flat_tensor/test/targets.bzl
@@ -35,8 +35,8 @@ def define_common_targets(is_fbcode=False):
             # The tests use this var to find the program file to load. This uses
             # an fbcode target path because the authoring/export tools
             # intentionally don't work in xplat (since they're host-only tools).
-            "ET_MODULE_LINEAR_PROGRAM": "$(location fbcode//executorch/test/models:exported_programs_with_data_separated[ModuleLinear.pte])",
-            "ET_MODULE_LINEAR_DATA": "$(location fbcode//executorch/test/models:exported_programs_with_data_separated[ModuleLinear.ptd])",
+            "ET_MODULE_LINEAR_PROGRAM_PATH": "$(location fbcode//executorch/test/models:exported_programs_with_data_separated[ModuleLinear.pte])",
+            "ET_MODULE_LINEAR_DATA_PATH": "$(location fbcode//executorch/test/models:exported_programs_with_data_separated[ModuleLinear.ptd])",
         }
 
         runtime.cxx_test(
diff --git a/test/models/targets.bzl b/test/models/targets.bzl
index e2cd0f264b3..95946a4bfab 100644
--- a/test/models/targets.bzl
+++ b/test/models/targets.bzl
@@ -95,7 +95,7 @@ def define_common_targets():
         name = "exported_programs_with_data_separated",
         cmd = "$(exe :export_program) --modules ModuleLinear --external-constants --outdir $OUT",
         outs = {
-            "ModuleLinear.pte": ["ModuleLinear.pte"],
+            "ModuleLinear.pte": ["ModuleLinearProgram.pte"],
             "ModuleLinear.ptd": ["_default_external_constant.ptd"],
         },
         default_outs = ["."],