diff --git a/batching-gson/src/test/java/com/flipkart/batching/gson/GsonSerializationTest.java b/batching-gson/src/test/java/com/flipkart/batching/gson/GsonSerializationTest.java
index aba8f59..620a174 100644
--- a/batching-gson/src/test/java/com/flipkart/batching/gson/GsonSerializationTest.java
+++ b/batching-gson/src/test/java/com/flipkart/batching/gson/GsonSerializationTest.java
@@ -37,7 +37,7 @@
 import org.json.JSONObject;
 import org.junit.Test;
 import org.junit.runner.RunWith;
-import org.robolectric.RobolectricGradleTestRunner;
+import org.robolectric.RobolectricTestRunner;
 import org.robolectric.annotation.Config;
 
 import java.io.IOException;
@@ -51,7 +51,7 @@
  * Test for {@link GsonSerializationStrategy}
  */
 
-@RunWith(RobolectricGradleTestRunner.class)
+@RunWith(RobolectricTestRunner.class)
 @Config(constants = BuildConfig.class, sdk = 21)
 public class GsonSerializationTest {