1 files changed, 2 insertions, 3 deletions
diff --git a/tests/robotests/src/com/android/tv/data/PreviewDataManagerTest.java b/tests/robotests/src/com/android/tv/data/PreviewDataManagerTest.java
index bfc8f7fa..b341abf4 100644
--- a/tests/robotests/src/com/android/tv/data/PreviewDataManagerTest.java
+++ b/tests/robotests/src/com/android/tv/data/PreviewDataManagerTest.java
@@ -35,10 +35,9 @@ import androidx.tvprovider.media.tv.TvContractCompat;
 import com.android.tv.data.api.Channel;
 import com.android.tv.data.api.Program;
 import com.android.tv.dvr.data.RecordedProgram;
+import com.android.tv.testing.TvRobolectricTestRunner;
 import com.android.tv.testing.constants.ConfigConstants;
 
-import org.robolectric.RobolectricTestRunner;
-
 import org.junit.Test;
 import org.junit.runner.RunWith;
 import org.robolectric.Robolectric;
@@ -50,7 +49,7 @@ import org.robolectric.shadows.ShadowLog;
 import java.util.List;
 
 /** Tests for {@link PreviewDataManager}. */
-@RunWith(RobolectricTestRunner.class)
+@RunWith(TvRobolectricTestRunner.class)
 @Config(sdk = ConfigConstants.SDK)
 public class PreviewDataManagerTest {
     private static final long FAKE_PREVIEW_CHANNEL_ID = 2002;