diff --git a/src/python/pants/backend/awslambda/python/rules_test.py b/src/python/pants/backend/awslambda/python/rules_test.py
index 768842536e9..828856de79f 100644
--- a/src/python/pants/backend/awslambda/python/rules_test.py
+++ b/src/python/pants/backend/awslambda/python/rules_test.py
@@ -71,6 +71,7 @@ def rule_runner() -> PythonRuleRunner:
             *package.rules(),
             QueryRule(BuiltPackage, (PythonAwsLambdaFieldSet,)),
             QueryRule(BuiltPackage, (PythonAwsLambdaLayerFieldSet,)),
+            QueryRule(BuiltPackage, (PexBinaryFieldSet,)),
         ],
         target_types=[
             FileTarget,
@@ -141,7 +142,6 @@ def complete_platform(rule_runner: PythonRuleRunner) -> bytes:
     pex_executable = os.path.join(rule_runner.build_root, "pex_exe/pex_exe.pex")
     return subprocess.run(
         args=[pex_executable, "interpreter", "inspect", "-mt"],
-        env=dict(PEX_MODULE="pex.cli", **os.environ),
         check=True,
         stdout=subprocess.PIPE,
     ).stdout
diff --git a/src/python/pants/backend/google_cloud_function/python/rules_test.py b/src/python/pants/backend/google_cloud_function/python/rules_test.py
index c4d30045637..2bc2e50ffb3 100644
--- a/src/python/pants/backend/google_cloud_function/python/rules_test.py
+++ b/src/python/pants/backend/google_cloud_function/python/rules_test.py
@@ -63,6 +63,7 @@ def rule_runner() -> PythonRuleRunner:
             *core_target_types_rules(),
             *package.rules(),
             QueryRule(BuiltPackage, (PythonGoogleCloudFunctionFieldSet,)),
+            QueryRule(BuiltPackage, (PexBinaryFieldSet,)),
         ],
         target_types=[
             FileTarget,
diff --git a/src/python/pants/backend/python/util_rules/pex_from_targets_test.py b/src/python/pants/backend/python/util_rules/pex_from_targets_test.py
index 800bf9b1299..685efb09511 100644
--- a/src/python/pants/backend/python/util_rules/pex_from_targets_test.py
+++ b/src/python/pants/backend/python/util_rules/pex_from_targets_test.py
@@ -521,7 +521,6 @@ def create_dists(
     find_links = workdir / "find-links"
     subprocess.run(
         args=[
-            pex_binary,
             pex_output,
             "repository",
             "extract",
@@ -529,7 +528,7 @@ def create_dists(
             find_links,
         ],
         check=True,
-        env=os.environ.copy() | {"PEX_MODULE": "pex.tools"},
+        env=os.environ.copy() | {"PEX_TOOLS": "1"},
     )
     return find_links