1 files changed, 3 insertions, 12 deletions
diff --git a/src/test/java/com/google/devtools/build/lib/syntax/EnvironmentTest.java b/src/test/java/com/google/devtools/build/lib/syntax/EnvironmentTest.java
index 64d293c5f0..3f01b28467 100644
--- a/src/test/java/com/google/devtools/build/lib/syntax/EnvironmentTest.java
+++ b/src/test/java/com/google/devtools/build/lib/syntax/EnvironmentTest.java
@@ -16,6 +16,7 @@ package com.google.devtools.build.lib.syntax;
 
 import static com.google.common.truth.Truth.assertThat;
 import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertNull;
 import static org.junit.Assert.fail;
 
 import com.google.common.collect.Sets;
@@ -39,12 +40,7 @@ public class EnvironmentTest extends EvaluationTestCase {
   // Test the API directly
   @Test
   public void testLookupAndUpdate() throws Exception {
-    try {
-      lookup("foo");
-      fail();
-    } catch (Environment.NoSuchVariableException e) {
-      assertThat(e).hasMessage("no such variable: foo");
-    }
+    assertNull(lookup("foo"));
     update("foo", "bar");
     assertEquals("bar", lookup("foo"));
   }
@@ -67,12 +63,7 @@ public class EnvironmentTest extends EvaluationTestCase {
   // Test assign through interpreter, lookup through API:
   @Test
   public void testAssign() throws Exception {
-    try {
-      lookup("foo");
-      fail();
-    } catch (Environment.NoSuchVariableException e) {
-      assertThat(e).hasMessage("no such variable: foo");
-    }
+    assertNull(lookup("foo"));
     eval("foo = 'bar'");
     assertEquals("bar", lookup("foo"));
   }