1 files changed, 3 insertions, 3 deletions
diff --git a/core/src/mill/define/Discover.scala b/core/src/mill/define/Discover.scala
index b213d9f3..fd5bd449 100644
--- a/core/src/mill/define/Discover.scala
+++ b/core/src/mill/define/Discover.scala
@@ -1,6 +1,6 @@
 package mill.define
 import language.experimental.macros
-import ammonite.main.Router.{EntryPoint, Overrides}
+import mill.main.Router.{EntryPoint, Overrides}
 import sourcecode.Compat.Context
 
 import scala.collection.mutable
@@ -41,7 +41,7 @@ object Discover {
     }
     rec(weakTypeOf[T])
 
-    val router = new mill.define.Router(c)
+    val router = new mill.main.Router(c)
     val mapping = for{
       discoveredModuleType <- seen
       val curCls = discoveredModuleType.asInstanceOf[router.c.Type]
@@ -57,7 +57,7 @@ object Discover {
       val (overrides, routes) = overridesRoutes.unzip
       val lhs =  q"classOf[${discoveredModuleType.typeSymbol.asClass}]"
       val clsType = discoveredModuleType.typeSymbol.asClass
-      val rhs = q"scala.Seq[(Int, ammonite.main.Router.EntryPoint[_])](..$overridesRoutes)"
+      val rhs = q"scala.Seq[(Int, mill.main.Router.EntryPoint[_])](..$overridesRoutes)"
       q"$lhs -> $rhs"
     }