diff --git a/build.gradle b/build.gradle
index 33409c5..c1f373e 100644
--- a/build.gradle
+++ b/build.gradle
@@ -70,7 +70,7 @@
     sourceSets {
         main {
             res.srcDirs = ['res']
-            java.srcDirs = ['src', 'src_shortcuts_overrides']
+            java.srcDirs = ['src', 'src_plugins']
             manifest.srcFile 'AndroidManifest-common.xml'
             proto {
                 srcDir 'protos/'
@@ -93,7 +93,7 @@
         }
 
         aosp {
-            java.srcDirs = ['src_flags', "src_ui_overrides"]
+            java.srcDirs = ['src_flags', "src_ui_overrides", 'src_shortcuts_overrides']
         }
 
         l3go {
@@ -104,7 +104,7 @@
 
         quickstep {
             res.srcDirs = ['quickstep/res']
-            java.srcDirs = ['src_flags', 'quickstep/src']
+            java.srcDirs = ['src_flags', 'quickstep/src', 'src_shortcuts_overrides']
             manifest.srcFile "quickstep/AndroidManifest.xml"
         }
     }