diff --git a/build.gradle b/build.gradle
index 3e7c5cbe..75d90942 100644
--- a/build.gradle
+++ b/build.gradle
@@ -85,6 +85,8 @@ checkstyle {
     toolVersion = "6.13"
 }
 
+checkstyleMain.source = "src/main/java"
+
 tasks.withType(JavaCompile) {
     options.compilerArgs += ["-Xlint:-options"]
 }
diff --git a/src/main/java/com/cdancy/bitbucket/rest/features/BranchApi.java b/src/main/java/com/cdancy/bitbucket/rest/features/BranchApi.java
index a4b6a35d..3d00feeb 100644
--- a/src/main/java/com/cdancy/bitbucket/rest/features/BranchApi.java
+++ b/src/main/java/com/cdancy/bitbucket/rest/features/BranchApi.java
@@ -28,13 +28,17 @@
 import org.jclouds.rest.annotations.Payload;
 import org.jclouds.rest.annotations.PayloadParam;
 import org.jclouds.rest.annotations.RequestFilters;
-
 import org.jclouds.rest.binders.BindToJsonPayload;
 
 import javax.inject.Named;
-
-import javax.ws.rs.*;
-
+import javax.ws.rs.Consumes;
+import javax.ws.rs.DELETE;
+import javax.ws.rs.GET;
+import javax.ws.rs.POST;
+import javax.ws.rs.PUT;
+import javax.ws.rs.Path;
+import javax.ws.rs.PathParam;
+import javax.ws.rs.Produces;
 import javax.ws.rs.core.MediaType;
 
 @Produces(MediaType.APPLICATION_JSON)