diff --git a/src/main/java/run/halo/app/controller/content/MainController.java b/src/main/java/run/halo/app/controller/content/MainController.java
index 5d1b9964a..fd01587ac 100644
--- a/src/main/java/run/halo/app/controller/content/MainController.java
+++ b/src/main/java/run/halo/app/controller/content/MainController.java
@@ -15,7 +15,7 @@ import javax.servlet.http.HttpServletResponse;
 import java.io.IOException;
 
 /**
- * Admin page.
+ * Main controller.
  *
  * @author ryanwang
  * @date : 2019-04-23
diff --git a/src/main/java/run/halo/app/model/freemarker/tag/CategoryTagDirective.java b/src/main/java/run/halo/app/model/freemarker/tag/CategoryTagDirective.java
index 50a2a07c0..44bd9d2de 100644
--- a/src/main/java/run/halo/app/model/freemarker/tag/CategoryTagDirective.java
+++ b/src/main/java/run/halo/app/model/freemarker/tag/CategoryTagDirective.java
@@ -45,6 +45,7 @@ public class CategoryTagDirective implements TemplateDirectiveModel {
                 case "listByPostId":
                     Integer postId = Integer.parseInt(params.get("postId").toString());
                     env.setVariable("categories", builder.build().wrap(postCategoryService.listCategoryBy(postId)));
+                    break;
                 case "count":
                     env.setVariable("count", builder.build().wrap(categoryService.count()));
                     break;