Updated the build.gradle files.

Almost, needed a little change.