Update build.gradle to properly run test