1 files changed, 2 insertions, 1 deletions
diff --git a/tests/src/com/android/inputmethod/latin/ShiftModeTests.java b/tests/src/com/android/inputmethod/latin/ShiftModeTests.java
index 59bb5f8a4..9b6e72be2 100644
--- a/tests/src/com/android/inputmethod/latin/ShiftModeTests.java
+++ b/tests/src/com/android/inputmethod/latin/ShiftModeTests.java
@@ -16,10 +16,11 @@
 
 package com.android.inputmethod.latin;
 
-import android.test.suitebuilder.annotation.LargeTest;
 import android.text.TextUtils;
 import android.view.inputmethod.EditorInfo;
 
+import androidx.test.filters.LargeTest;
+
 import com.android.inputmethod.latin.common.Constants;
 
 @LargeTest