diff --git a/gradle-scalastyle-plugin_2.10/src/main/groovy/org/github/ngbinh/scalastyle/ScalaStyleTask.groovy b/gradle-scalastyle-plugin_2.10/src/main/groovy/org/github/ngbinh/scalastyle/ScalaStyleTask.groovy
index 01c07c8..3412330 100644
--- a/gradle-scalastyle-plugin_2.10/src/main/groovy/org/github/ngbinh/scalastyle/ScalaStyleTask.groovy
+++ b/gradle-scalastyle-plugin_2.10/src/main/groovy/org/github/ngbinh/scalastyle/ScalaStyleTask.groovy
@@ -20,13 +20,9 @@ package org.github.ngbinh.scalastyle
 import org.gradle.api.file.FileTree
 import org.gradle.api.tasks.SourceTask
 import org.gradle.api.tasks.TaskAction
-import org.scalastyle.ScalastyleChecker
 import org.scalastyle.ScalastyleConfiguration
 import org.scalastyle.TextOutput
 import org.scalastyle.XmlOutput
-import com.typesafe.config.ConfigFactory
-import com.typesafe.config.Config
-
 /**
  * @author Binh Nguyen
  * @since 12/16/2014
@@ -68,7 +64,7 @@ class ScalaStyleTask extends SourceTask {
                 if (testConfigLocation != null) {
                   def testConfiguration = ScalastyleConfiguration.readFromXml(testConfigLocation)
                   if (testConfiguration != null) {
-                      def testFilesToProcess = scalaStyleUtils.getTestFilesToProcess(source.getFiles().toList(), testSourceDir.getFiles().toList(), inputEncoding, includeTestSourceDirectory)
+                      def testFilesToProcess = scalaStyleUtils.getTestFilesToProcess(testSourceDir.getFiles().toList(), inputEncoding)
                       messages.addAll(scalaStyleUtils.checkFiles(testConfiguration, testFilesToProcess))
                   }
                 }
@@ -120,7 +116,7 @@ class ScalaStyleTask extends SourceTask {
             throw new Exception("Specify Scala source set")
         }
 
-        if (includeTestSourceDirectory && testSource == null) {
+        if (testSource == null) {
             testSourceDir = project.fileTree(project.projectDir.absolutePath + "/src/test/scala")
         } else {
             testSourceDir = project.fileTree(project.projectDir.absolutePath + "/" + testSource)
diff --git a/gradle-scalastyle-plugin_2.10/src/main/scala/org/github/ngbinh/scalastyle/ScalaStyleUtils.scala b/gradle-scalastyle-plugin_2.10/src/main/scala/org/github/ngbinh/scalastyle/ScalaStyleUtils.scala
index ae2b486..db308db 100644
--- a/gradle-scalastyle-plugin_2.10/src/main/scala/org/github/ngbinh/scalastyle/ScalaStyleUtils.scala
+++ b/gradle-scalastyle-plugin_2.10/src/main/scala/org/github/ngbinh/scalastyle/ScalaStyleUtils.scala
@@ -41,7 +41,7 @@ class ScalaStyleUtils {
     sd ::: tsd
   }
 
-  def getTestFilesToProcess(testFiles: jList[File], inputEncoding: String, includeTestSourceDirectory: Boolean): List[FileSpec] = {
+  def getTestFilesToProcess(testFiles: jList[File], inputEncoding: String): List[FileSpec] = {
     getFiles("testFiles", asScalaBufferConverter(testFiles).asScala.toList, inputEncoding)
   }
 
diff --git a/gradle-scalastyle-plugin_2.11/src/main/groovy/org/github/ngbinh/scalastyle/ScalaStyleTask.groovy b/gradle-scalastyle-plugin_2.11/src/main/groovy/org/github/ngbinh/scalastyle/ScalaStyleTask.groovy
index 01c07c8..3412330 100644
--- a/gradle-scalastyle-plugin_2.11/src/main/groovy/org/github/ngbinh/scalastyle/ScalaStyleTask.groovy
+++ b/gradle-scalastyle-plugin_2.11/src/main/groovy/org/github/ngbinh/scalastyle/ScalaStyleTask.groovy
@@ -20,13 +20,9 @@ package org.github.ngbinh.scalastyle
 import org.gradle.api.file.FileTree
 import org.gradle.api.tasks.SourceTask
 import org.gradle.api.tasks.TaskAction
-import org.scalastyle.ScalastyleChecker
 import org.scalastyle.ScalastyleConfiguration
 import org.scalastyle.TextOutput
 import org.scalastyle.XmlOutput
-import com.typesafe.config.ConfigFactory
-import com.typesafe.config.Config
-
 /**
  * @author Binh Nguyen
  * @since 12/16/2014
@@ -68,7 +64,7 @@ class ScalaStyleTask extends SourceTask {
                 if (testConfigLocation != null) {
                   def testConfiguration = ScalastyleConfiguration.readFromXml(testConfigLocation)
                   if (testConfiguration != null) {
-                      def testFilesToProcess = scalaStyleUtils.getTestFilesToProcess(source.getFiles().toList(), testSourceDir.getFiles().toList(), inputEncoding, includeTestSourceDirectory)
+                      def testFilesToProcess = scalaStyleUtils.getTestFilesToProcess(testSourceDir.getFiles().toList(), inputEncoding)
                       messages.addAll(scalaStyleUtils.checkFiles(testConfiguration, testFilesToProcess))
                   }
                 }
@@ -120,7 +116,7 @@ class ScalaStyleTask extends SourceTask {
             throw new Exception("Specify Scala source set")
         }
 
-        if (includeTestSourceDirectory && testSource == null) {
+        if (testSource == null) {
             testSourceDir = project.fileTree(project.projectDir.absolutePath + "/src/test/scala")
         } else {
             testSourceDir = project.fileTree(project.projectDir.absolutePath + "/" + testSource)
diff --git a/gradle-scalastyle-plugin_2.11/src/main/scala/org/github/ngbinh/scalastyle/ScalaStyleUtils.scala b/gradle-scalastyle-plugin_2.11/src/main/scala/org/github/ngbinh/scalastyle/ScalaStyleUtils.scala
index ae2b486..db308db 100644
--- a/gradle-scalastyle-plugin_2.11/src/main/scala/org/github/ngbinh/scalastyle/ScalaStyleUtils.scala
+++ b/gradle-scalastyle-plugin_2.11/src/main/scala/org/github/ngbinh/scalastyle/ScalaStyleUtils.scala
@@ -41,7 +41,7 @@ class ScalaStyleUtils {
     sd ::: tsd
   }
 
-  def getTestFilesToProcess(testFiles: jList[File], inputEncoding: String, includeTestSourceDirectory: Boolean): List[FileSpec] = {
+  def getTestFilesToProcess(testFiles: jList[File], inputEncoding: String): List[FileSpec] = {
     getFiles("testFiles", asScalaBufferConverter(testFiles).asScala.toList, inputEncoding)
   }