diff --git a/uhabits-android/build.gradle b/uhabits-android/build.gradle
index 0a8170e61..bd78b5ba9 100644
--- a/uhabits-android/build.gradle
+++ b/uhabits-android/build.gradle
@@ -1,3 +1,4 @@
+apply plugin: 'idea'
 apply plugin: 'com.android.application'
 apply plugin: 'jacoco'
 
@@ -81,6 +82,10 @@ dependencies {
     androidTestAnnotationProcessor 'com.google.auto.factory:auto-factory:1.0-beta3'
     androidTestAnnotationProcessor 'com.jakewharton:butterknife-compiler:8.6.1-SNAPSHOT'
 
+    testAnnotationProcessor 'com.google.dagger:dagger-compiler:2.11-rc2'
+    testAnnotationProcessor 'com.google.auto.factory:auto-factory:1.0-beta3'
+    testAnnotationProcessor 'com.jakewharton:butterknife-compiler:8.6.1-SNAPSHOT'
+
     implementation('com.opencsv:opencsv:3.9') {
         exclude group: 'commons-logging', module: 'commons-logging'
     }
diff --git a/uhabits-core/build.gradle b/uhabits-core/build.gradle
index fa1ec0e0b..47f7cc009 100644
--- a/uhabits-core/build.gradle
+++ b/uhabits-core/build.gradle
@@ -1,11 +1,19 @@
+plugins {
+    id "net.ltgt.apt" version "0.7"
+}
+apply plugin: 'idea'
 apply plugin: 'java'
 apply plugin: 'jacoco'
 
 dependencies {
 
+    apt 'com.google.auto.factory:auto-factory:1.0-beta3'
+    apt 'com.google.dagger:dagger:2.11-rc2'
+
     compileOnly 'javax.annotation:jsr250-api:1.0'
     compileOnly 'com.google.auto.factory:auto-factory:1.0-beta3'
-    implementation 'com.google.dagger:dagger:2.11-rc2'
+    compileOnly 'com.google.dagger:dagger:2.11-rc2'
+
     implementation 'com.android.support:support-annotations:25.3.1'
     implementation 'com.google.code.findbugs:jsr305:3.0.2'
     implementation 'org.apache.commons:commons-lang3:3.5'