diff --git a/tests/robotests/src/com/android/settings/testutils/BatteryTestUtils.java b/tests/robotests/src/com/android/settings/testutils/BatteryTestUtils.java
index b75fa8cfe1c..c54b750cb55 100644
--- a/tests/robotests/src/com/android/settings/testutils/BatteryTestUtils.java
+++ b/tests/robotests/src/com/android/settings/testutils/BatteryTestUtils.java
@@ -190,6 +190,7 @@ public class BatteryTestUtils {
         when(mockUsbPort.getStatus()).thenReturn(mockUsbPortStatus);
         when(mockUsbPort.supportsComplianceWarnings()).thenReturn(true);
         when(mockUsbPortStatus.isConnected()).thenReturn(true);
-        when(mockUsbPortStatus.getComplianceWarnings()).thenReturn(new int[]{1});
+        when(mockUsbPortStatus.getComplianceWarnings())
+                .thenReturn(new int[]{UsbPortStatus.COMPLIANCE_WARNING_OTHER});
     }
 }