1 files changed, 28 insertions, 21 deletions
diff --git a/doc-tool/src/dotty/tools/dottydoc/staticsite/DefaultParams.scala b/doc-tool/src/dotty/tools/dottydoc/staticsite/DefaultParams.scala
index 066b23e0b..a53f6f0d4 100644
--- a/doc-tool/src/dotty/tools/dottydoc/staticsite/DefaultParams.scala
+++ b/doc-tool/src/dotty/tools/dottydoc/staticsite/DefaultParams.scala
@@ -17,27 +17,34 @@ case class DefaultParams(
 ) {
   import model.JavaConverters._
 
-  def toMap: Map[String, AnyRef] = Map(
-    "docs" -> docs,
-
-    "originalDocs" -> originalDocs,
-
-    "page" -> Map(
-      "url" -> page.url,
-      "date" -> page.date,
-      "path" -> page.path
-    ),
-
-    "site" -> Map(
-      "baseurl" -> site.baseurl,
-      "posts" -> site.posts.map(_.toMap),
-      "project" -> site.projectTitle
-    ).asJava,
-
-    "sidebar" -> sidebar.titles.asJava,
-
-    "entity" -> entity.asJava()
-  )
+  def toMap: Map[String, AnyRef] = {
+    val base = Map(
+      "docs" -> docs,
+
+      "originalDocs" -> originalDocs,
+
+      "page" -> Map(
+        "url" -> page.url,
+        "date" -> page.date,
+        "path" -> page.path
+      ),
+
+      "site" -> Map(
+        "baseurl" -> site.baseurl,
+        "posts" -> site.posts.map(_.toMap),
+        "project" -> site.projectTitle
+      ).asJava,
+
+      "sidebar" -> sidebar.titles.asJava
+    )
+    val entityMap = entity match {
+      case NonEntity => Map.empty
+      case _ => Map(
+        "entity" -> entity.asJava
+      )
+    }
+    base ++ entityMap
+  }
 
   def withPosts(posts: Array[BlogPost]): DefaultParams =
     copy(site = SiteInfo(site.baseurl, site.projectTitle, posts))