diff --git a/modules/kobalt-plugin-api/src/main/kotlin/com/beust/kobalt/TestDirective.kt b/modules/kobalt-plugin-api/src/main/kotlin/com/beust/kobalt/TestDirective.kt
index 31052002..f62a9098 100644
--- a/modules/kobalt-plugin-api/src/main/kotlin/com/beust/kobalt/TestDirective.kt
+++ b/modules/kobalt-plugin-api/src/main/kotlin/com/beust/kobalt/TestDirective.kt
@@ -14,10 +14,6 @@ class TestConfig(val project: Project) {
 }
 
 @Directive
-fun Project.test(init: TestConfig.() -> Unit) : TestConfig {
-    val result = TestConfig(this)
-    result.init()
-    return result
-}
+fun Project.test(init: TestConfig.() -> Unit) = TestConfig(this).apply { init() }