From e0770d64543f42fe0957d318553fcb232ec838d1 Mon Sep 17 00:00:00 2001
From: Andrii Kulian <akulian@google.com>
Date: Thu, 23 Sep 2021 18:34:32 -0700
Subject: [PATCH] Fix FlakyTest import

Bug: 200332057
Test: Treehugger
Change-Id: I557d7c4b928127b6d92782a3b5cf1a11fcdfaa1b
---
 .../server/devicestate/DeviceStateManagerServiceTest.java       | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/services/tests/servicestests/src/com/android/server/devicestate/DeviceStateManagerServiceTest.java b/services/tests/servicestests/src/com/android/server/devicestate/DeviceStateManagerServiceTest.java
index 3fcd38b6771a1..03eba9bfc35c2 100644
--- a/services/tests/servicestests/src/com/android/server/devicestate/DeviceStateManagerServiceTest.java
+++ b/services/tests/servicestests/src/com/android/server/devicestate/DeviceStateManagerServiceTest.java
@@ -34,10 +34,10 @@ import android.hardware.devicestate.IDeviceStateManagerCallback;
 import android.os.Binder;
 import android.os.IBinder;
 import android.os.RemoteException;
-import android.platform.test.annotations.FlakyTest;
 import android.platform.test.annotations.Presubmit;
 
 import androidx.test.InstrumentationRegistry;
+import androidx.test.filters.FlakyTest;
 import androidx.test.runner.AndroidJUnit4;
 
 import com.android.server.wm.ActivityTaskManagerInternal;