diff --git a/src/main/java/com/itfsw/mybatis/generator/plugins/LogicalDeletePlugin.java b/src/main/java/com/itfsw/mybatis/generator/plugins/LogicalDeletePlugin.java
index d7ef0c0..0cc661a 100644
--- a/src/main/java/com/itfsw/mybatis/generator/plugins/LogicalDeletePlugin.java
+++ b/src/main/java/com/itfsw/mybatis/generator/plugins/LogicalDeletePlugin.java
@@ -331,8 +331,6 @@ public class LogicalDeletePlugin extends BasePlugin {
 
                 selectByPrimaryKey.addAttribute(new Attribute("parameterType", "map"));
 
-                context.getCommentGenerator().addComment(selectByPrimaryKey);
-
                 sb = new StringBuilder();
                 sb.append("select ");
 
diff --git a/src/main/java/com/itfsw/mybatis/generator/plugins/utils/FormatTools.java b/src/main/java/com/itfsw/mybatis/generator/plugins/utils/FormatTools.java
index e93782f..d108ddf 100644
--- a/src/main/java/com/itfsw/mybatis/generator/plugins/utils/FormatTools.java
+++ b/src/main/java/com/itfsw/mybatis/generator/plugins/utils/FormatTools.java
@@ -215,7 +215,7 @@ public class FormatTools {
         boolean flag = false;
         while (elementIterator.hasNext()) {
             Element ele = elementIterator.next();
-            if (ele instanceof TextElement && ((TextElement) ele).getContent().matches("<!--")) {
+            if (ele instanceof TextElement && ((TextElement) ele).getContent().matches(".*<!--.*")) {
                 flag = true;
             }
 
@@ -223,7 +223,7 @@ public class FormatTools {
                 elementIterator.remove();
             }
 
-            if (ele instanceof TextElement && ((TextElement) ele).getContent().matches("-->")) {
+            if (ele instanceof TextElement && ((TextElement) ele).getContent().matches(".*-->.*")) {
                 flag = false;
             }
         }