diff --git a/Privileged-Extension/build.gradle b/Privileged-Extension/build.gradle
index 6e171e434..1b3cdbe3c 100644
--- a/Privileged-Extension/build.gradle
+++ b/Privileged-Extension/build.gradle
@@ -36,14 +36,13 @@ android {
 }
 
 checkstyle {
-    toolVersion = '6.17'
+    toolVersion = '6.18'
 }
 
 task checkstyle(type: Checkstyle) {
     configFile file("${project.rootDir}/config/checkstyle/checkstyle.xml")
-    source 'src'
+    source 'src/main/java'
     include '**/*.java'
-    exclude '**/gen/**'
 
     classpath = files()
 }
diff --git a/app/build.gradle b/app/build.gradle
index 7442af8bf..5103af08b 100644
--- a/app/build.gradle
+++ b/app/build.gradle
@@ -191,7 +191,7 @@ android {
 }
 
 checkstyle {
-    toolVersion = '6.17'
+    toolVersion = '6.18'
 }
 
 task checkstyle(type: Checkstyle) {