From 14d002ed967f93a55e4350b6ed56258291d815fd Mon Sep 17 00:00:00 2001
From: Li Haoyi <haoyi.sg@gmail.com>
Date: Thu, 17 May 2018 22:40:39 -0700
Subject: [PATCH] Update ScalaModule.scala

---
 scalalib/src/mill/scalalib/ScalaModule.scala | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/scalalib/src/mill/scalalib/ScalaModule.scala b/scalalib/src/mill/scalalib/ScalaModule.scala
index e54e2e02e6c..921cc3bdd76 100644
--- a/scalalib/src/mill/scalalib/ScalaModule.scala
+++ b/scalalib/src/mill/scalalib/ScalaModule.scala
@@ -125,7 +125,7 @@ trait ScalaModule extends JavaModule { outer =>
     val files = for{
       ref <- allSources()
       if exists(ref.path)
-      p <- ls.rec(ref.path)
+      p <- (if (ref.path.isDir) ls.rec(ref.path) else Seq(ref.path))
       if (p.isFile && ((p.ext == "scala") || (p.ext == "java")))
     } yield p.toNIO.toString