diff --git a/build.gradle b/build.gradle
index 83acd6d..9f973c1 100644
--- a/build.gradle
+++ b/build.gradle
@@ -19,12 +19,16 @@
         jcenter()
     }
     dependencies {
-        classpath 'com.android.tools.build:gradle:1.2.3'
+        classpath 'com.android.tools.build:gradle:1.3.1'
     }
 }
 
 apply plugin: 'com.android.library'
 
+repositories {
+    jcenter()
+}
+
 android {
     compileSdkVersion 22
     buildToolsVersion = '22.0.1'
diff --git a/rules.gradle b/rules.gradle
index 40508d0..04dd681 100644
--- a/rules.gradle
+++ b/rules.gradle
@@ -1,3 +1,12 @@
 // See build.gradle for an explanation of what this file is.
 
 apply plugin: 'com.android.library'
+
+// Check if the android plugin version supports unit testing.
+if (configurations.findByName("testCompile")) {
+  dependencies {
+    testCompile "junit:junit:4.10"
+    testCompile "org.mockito:mockito-core:1.9.5"
+    testCompile "org.robolectric:robolectric:3.0"
+  }
+}