diff --git a/extensions/ima/build.gradle b/extensions/ima/build.gradle
index 3a20e378ae..1a35ad3450 100644
--- a/extensions/ima/build.gradle
+++ b/extensions/ima/build.gradle
@@ -21,7 +21,7 @@ android {
     defaultConfig {
         minSdkVersion project.ext.minSdkVersion
         targetSdkVersion project.ext.targetSdkVersion
-        testInstrumentationRunner "android.support.test.runner.AndroidJUnitRunner"
+        consumerProguardFiles 'proguard-rules.txt'
     }
 }