diff --git a/tests/robotests/src/com/android/settings/accessibility/shortcuts/EditShortcutsPreferenceFragmentTest.java b/tests/robotests/src/com/android/settings/accessibility/shortcuts/EditShortcutsPreferenceFragmentTest.java
index b2ddb6c3cd5..56486d2a871 100644
--- a/tests/robotests/src/com/android/settings/accessibility/shortcuts/EditShortcutsPreferenceFragmentTest.java
+++ b/tests/robotests/src/com/android/settings/accessibility/shortcuts/EditShortcutsPreferenceFragmentTest.java
@@ -549,6 +549,7 @@ public class EditShortcutsPreferenceFragmentTest {
             ComponentName componentName, String label) {
         AccessibilityTarget target = mock(AccessibilityTarget.class);
         when(target.getComponentName()).thenReturn(componentName);
+        when(target.getId()).thenReturn(componentName.flattenToString());
         when(target.getLabel()).thenReturn(label);
         return target;
     }