From c948427e2c11b412ab7121c4c2804105cde711ae Mon Sep 17 00:00:00 2001
From: Li Haoyi <haoyi.sg@gmail.com>
Date: Sat, 20 Oct 2018 13:07:35 +0800
Subject: 0.3.2

---
 readme.md                                   | 2 +-
 scalalib/src/mill/scalalib/JavaModule.scala | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/readme.md b/readme.md
index c4f1da0f..634613bf 100644
--- a/readme.md
+++ b/readme.md
@@ -151,7 +151,7 @@ optimizer without classpath conflicts.
 
 ## Changelog
 
-### 0.3.1
+### 0.3.2
 
 - Automatically detect main class to make `ScalaModule#assembly` self-executable
 
diff --git a/scalalib/src/mill/scalalib/JavaModule.scala b/scalalib/src/mill/scalalib/JavaModule.scala
index f1df22a6..111cfdb2 100644
--- a/scalalib/src/mill/scalalib/JavaModule.scala
+++ b/scalalib/src/mill/scalalib/JavaModule.scala
@@ -161,7 +161,7 @@ trait JavaModule extends mill.Module with TaskModule { outer =>
     * OS-X and Linux
     */
   def prependShellScript: T[String] = T{
-    mainClass() match{
+    finalMainClassOpt().toOption match{
       case None => ""
       case Some(cls) =>
         val isWin = scala.util.Properties.isWin
-- 
cgit v1.2.3