Merge "Put back annotations in DeviceStateRotationLockSettingControllerTest" into main
diff --git a/packages/SystemUI/tests/src/com/android/systemui/statusbar/policy/DeviceStateRotationLockSettingControllerTest.java b/packages/SystemUI/tests/src/com/android/systemui/statusbar/policy/DeviceStateRotationLockSettingControllerTest.java
index 564dca0..3247a1a 100644
--- a/packages/SystemUI/tests/src/com/android/systemui/statusbar/policy/DeviceStateRotationLockSettingControllerTest.java
+++ b/packages/SystemUI/tests/src/com/android/systemui/statusbar/policy/DeviceStateRotationLockSettingControllerTest.java
@@ -354,8 +354,8 @@
     }
 
     private static DeviceState createDeviceState(int identifier, @NonNull String name,
-            @NonNull Set<Integer> systemProperties,
-            @NonNull Set<Integer> physicalProperties) {
+            @NonNull Set<@DeviceState.SystemDeviceStateProperties Integer> systemProperties,
+            @NonNull Set<@DeviceState.PhysicalDeviceStateProperties Integer> physicalProperties) {
         DeviceState.Configuration deviceStateConfiguration = new DeviceState.Configuration.Builder(
                 identifier, name).setSystemProperties(systemProperties).setPhysicalProperties(
                 physicalProperties).build();