diff --git a/tests/robotests/src/com/android/settings/display/BrightnessLevelPreferenceControllerTest.java b/tests/robotests/src/com/android/settings/display/BrightnessLevelPreferenceControllerTest.java
index c2de51050e8..c2a8611ec02 100644
--- a/tests/robotests/src/com/android/settings/display/BrightnessLevelPreferenceControllerTest.java
+++ b/tests/robotests/src/com/android/settings/display/BrightnessLevelPreferenceControllerTest.java
@@ -153,7 +153,7 @@ public class BrightnessLevelPreferenceControllerTest {
 
         when(mDisplay.getBrightnessInfo()).thenReturn(
                 new BrightnessInfo(0.1f, 0.0f, 1.0f, BrightnessInfo.HIGH_BRIGHTNESS_MODE_OFF,
-                    0.5f));
+                    0.5f, BrightnessInfo.BRIGHTNESS_MAX_REASON_NONE));
 
         mController.updateState(mPreference);
 
@@ -168,7 +168,7 @@ public class BrightnessLevelPreferenceControllerTest {
 
         when(mDisplay.getBrightnessInfo()).thenReturn(
                 new BrightnessInfo(0.5f, 0.0f, 1.0f, BrightnessInfo.HIGH_BRIGHTNESS_MODE_OFF,
-                    0.5f));
+                    0.5f, BrightnessInfo.BRIGHTNESS_MAX_REASON_NONE));
 
         mController.updateState(mPreference);