diff --git a/java/build.gradle b/java/build.gradle
index 36d9d11f50..532dfea6ac 100644
--- a/java/build.gradle
+++ b/java/build.gradle
@@ -23,6 +23,8 @@ dependencies {
     compile 'org.whispersystems:axolotl-java:1.1.1'
     compile 'com.squareup.okhttp:okhttp:2.2.0'
     compile 'org.apache.httpcomponents:httpclient:4.4'
+
+    testCompile 'junit:junit:3.8.2'
 }
 
 
diff --git a/java/src/androidTest/java/org/whispersystems/textsecure/push/PushTransportDetailsTest.java b/java/src/test/java/org/whispersystems/textsecure/push/PushTransportDetailsTest.java
similarity index 90%
rename from java/src/androidTest/java/org/whispersystems/textsecure/push/PushTransportDetailsTest.java
rename to java/src/test/java/org/whispersystems/textsecure/push/PushTransportDetailsTest.java
index 4d6c5ab32d..72686ac775 100644
--- a/java/src/androidTest/java/org/whispersystems/textsecure/push/PushTransportDetailsTest.java
+++ b/java/src/test/java/org/whispersystems/textsecure/push/PushTransportDetailsTest.java
@@ -1,10 +1,10 @@
 package org.whispersystems.textsecure.push;
 
-import android.test.AndroidTestCase;
+import junit.framework.TestCase;
 
 import org.whispersystems.textsecure.internal.push.PushTransportDetails;
 
-public class PushTransportDetailsTest extends AndroidTestCase {
+public class PushTransportDetailsTest extends TestCase {
 
   private final PushTransportDetails transportV2 = new PushTransportDetails(2);
   private final PushTransportDetails transportV3 = new PushTransportDetails(3);