1 files changed, 3 insertions, 4 deletions
diff --git a/src/test/java/com/google/devtools/build/lib/skylark/SkylarkRuleImplementationFunctionsTest.java b/src/test/java/com/google/devtools/build/lib/skylark/SkylarkRuleImplementationFunctionsTest.java
index af8b8aa2e6..8dfb22dc5a 100644
--- a/src/test/java/com/google/devtools/build/lib/skylark/SkylarkRuleImplementationFunctionsTest.java
+++ b/src/test/java/com/google/devtools/build/lib/skylark/SkylarkRuleImplementationFunctionsTest.java
@@ -1563,11 +1563,10 @@ public class SkylarkRuleImplementationFunctionsTest extends SkylarkTestCase {
 
   @Test
   public void testDefinedMakeVariable() throws Exception {
+    useConfiguration("--define=FOO=bar");
     SkylarkRuleContext ctx = createRuleContext("//foo:baz");
-    String java = (String) evalRuleContextCode(ctx, "ruleContext.var['JAVA']");
-    // Get the last path segment
-    java = java.substring(java.lastIndexOf('/'));
-    assertThat(java).isEqualTo("/java" + OsUtils.executableExtension());
+    String foo = (String) evalRuleContextCode(ctx, "ruleContext.var['FOO']");
+    assertThat(foo).isEqualTo("bar");
   }
 
   @Test