1 files changed, 5 insertions, 2 deletions
diff --git a/scalaplugin/src/main/scala/mill/scalaplugin/ScalaModule.scala b/scalaplugin/src/main/scala/mill/scalaplugin/ScalaModule.scala
index d2c37782..5a1f261a 100644
--- a/scalaplugin/src/main/scala/mill/scalaplugin/ScalaModule.scala
+++ b/scalaplugin/src/main/scala/mill/scalaplugin/ScalaModule.scala
@@ -7,7 +7,7 @@ import java.util.Optional
 import ammonite.ops._
 import coursier.{Cache, Fetch, MavenRepository, Repository, Resolution}
 import mill.define.Task
-import mill.define.Task.Module
+import mill.define.Task.{Module, TaskModule}
 import mill.eval.{Evaluator, PathRef}
 import mill.modules.Jvm.{createAssembly, createJar}
 import mill.util.OSet
@@ -161,7 +161,7 @@ object ScalaModule{
   )
 }
 import ScalaModule._
-trait TestScalaModule extends ScalaModule{
+trait TestScalaModule extends ScalaModule with TaskModule{
   def testFramework: T[String]
   def run() = T.command{
     TestRunner(
@@ -170,6 +170,7 @@ trait TestScalaModule extends ScalaModule{
       Seq(compile().path)
     )
   }
+  def self() = run()
 }
 trait ScalaModule extends Module{ outer =>
   trait Tests extends TestScalaModule{
@@ -267,6 +268,8 @@ trait ScalaModule extends Module{ outer =>
     PathRef(dest)
   }
 
+
+
   def run(mainClass: String) = T.command{
     import ammonite.ops._, ImplicitWd._
     %('java, "-cp", (runDepClasspath().map(_.path) :+ compile().path).mkString(":"), mainClass)