1 files changed, 4 insertions, 2 deletions
diff --git a/src/main/groovy/FlycheckAndroidJavaTask.groovy b/src/main/groovy/FlycheckAndroidJavaTask.groovy
index 7fef801..e8ebf4f 100644
--- a/src/main/groovy/FlycheckAndroidJavaTask.groovy
+++ b/src/main/groovy/FlycheckAndroidJavaTask.groovy
@@ -18,9 +18,11 @@ class FlycheckAndroidJavaTask extends DefaultTask {
         } else if (android && android.class.simpleName == 'LibraryExtension_Decorated') {
             variants = android.libraryVariants.collect { it.name }
         } else {
-            def buildTypes = project.extensions.buildTypes.names
+            def buildTypesExt = project.extensions.findByName('buildTypes')
+            def flavorsExt = project.extensions.findByName('flavors')
+            def buildTypes = buildTypesExt ? buildTypesExt.names : []
             if (buildTypes.isEmpty()) buildTypes = ['debug', 'release']
-            def flavors = project.extensions.flavors.names
+            def flavors = flavorsExt ? flavorsExt.names : []
             if (flavors.isEmpty()) {
                 variants = buildTypes
             } else {