diff --git a/kobalt/src/Build.kt b/kobalt/src/Build.kt
index 107ca47d..218e9f79 100644
--- a/kobalt/src/Build.kt
+++ b/kobalt/src/Build.kt
@@ -1,7 +1,6 @@
 
 import com.beust.kobalt.*
 import com.beust.kobalt.api.*
-import com.beust.kobalt.internal.*
 import com.beust.kobalt.plugin.application.application
 import com.beust.kobalt.plugin.java.javaCompiler
 import com.beust.kobalt.plugin.java.javaProject
diff --git a/src/main/kotlin/com/beust/kobalt/TestDirective.kt b/src/main/kotlin/com/beust/kobalt/TestDirective.kt
new file mode 100644
index 00000000..e8f80fed
--- /dev/null
+++ b/src/main/kotlin/com/beust/kobalt/TestDirective.kt
@@ -0,0 +1,19 @@
+package com.beust.kobalt
+
+import com.beust.kobalt.api.Project
+import com.beust.kobalt.api.annotation.Directive
+
+class TestConfig(val project: Project) {
+    fun args(vararg arg: String) {
+        project.testArgs.addAll(arg)
+    }
+}
+
+@Directive
+fun Project.test(init: TestConfig.() -> Unit) : TestConfig {
+    val result = TestConfig(this)
+    result.init()
+    return result
+}
+
+
diff --git a/src/main/kotlin/com/beust/kobalt/internal/JvmCompilerPlugin.kt b/src/main/kotlin/com/beust/kobalt/internal/JvmCompilerPlugin.kt
index d6221160..988482f8 100644
--- a/src/main/kotlin/com/beust/kobalt/internal/JvmCompilerPlugin.kt
+++ b/src/main/kotlin/com/beust/kobalt/internal/JvmCompilerPlugin.kt
@@ -4,7 +4,6 @@ import com.beust.kobalt.TaskResult
 import com.beust.kobalt.api.BasePlugin
 import com.beust.kobalt.api.KobaltContext
 import com.beust.kobalt.api.Project
-import com.beust.kobalt.api.annotation.Directive
 import com.beust.kobalt.api.annotation.ExportedProperty
 import com.beust.kobalt.api.annotation.Task
 import com.beust.kobalt.maven.*
@@ -126,17 +125,3 @@ abstract class JvmCompilerPlugin @Inject constructor(
         compilerArgs.addAll(args)
     }
 }
-
-class TestConfig(val project: Project) {
-    fun args(vararg arg: String) {
-        project.testArgs.addAll(arg)
-    }
-}
-
-@Directive
-fun Project.test(init: TestConfig.() -> Unit) : TestConfig {
-    val result = TestConfig(this)
-    result.init()
-    return result
-}
-