diff --git a/wpilibc/build.gradle b/wpilibc/build.gradle
index 8859eea9bf..44f1bf9d2d 100644
--- a/wpilibc/build.gradle
+++ b/wpilibc/build.gradle
@@ -78,8 +78,8 @@ model {
                 cpp {
                     def dir = 'wpilibC++IntegrationTests'
                     source {
-                        srcDir "${dir}/src"
-                        include '*.cpp'
+                        srcDirs = ["${dir}/src", "${dir}/src/command"]
+                        includes = ['*.cpp']
                     }
                     source {
                         srcDir "${dir}/src/gtest/src"