1 files changed, 4 insertions, 0 deletions
diff --git a/tests/src/com/android/inputmethod/latin/BinaryDictionaryTests.java b/tests/src/com/android/inputmethod/latin/BinaryDictionaryTests.java
index a26c25886..65c70a6db 100644
--- a/tests/src/com/android/inputmethod/latin/BinaryDictionaryTests.java
+++ b/tests/src/com/android/inputmethod/latin/BinaryDictionaryTests.java
@@ -993,6 +993,10 @@ public class BinaryDictionaryTests extends AndroidTestCase {
         for (final String word : words) {
             final UnigramProperty unigramProperty = binaryDictionary.getUnigramProperty(word);
             assertEquals((int)unigramProbabilities.get(word), unigramProperty.mProbability);
+            if (!shortcutTargets.containsKey(word)) {
+                // The word does not have shortcut targets.
+                continue;
+            }
             assertEquals(shortcutTargets.get(word).size(), unigramProperty.mShortcutTargets.size());
             for (final WeightedString shortcutTarget : unigramProperty.mShortcutTargets) {
                 final String targetCodePonts = shortcutTarget.mWord;