diff --git a/server/like-generator/src/main/java/com/hxkj/generator/service/impl/GenerateServiceImpl.java b/server/like-generator/src/main/java/com/hxkj/generator/service/impl/GenerateServiceImpl.java
index c4ca61bd..6c074072 100644
--- a/server/like-generator/src/main/java/com/hxkj/generator/service/impl/GenerateServiceImpl.java
+++ b/server/like-generator/src/main/java/com/hxkj/generator/service/impl/GenerateServiceImpl.java
@@ -373,7 +373,7 @@ public class GenerateServiceImpl implements IGenerateService {
             StringWriter sw = new StringWriter();
             Template tpl = Velocity.getTemplate(template, GenConstants.UTF8);
             tpl.merge(context, sw);
-            map.put(template, sw.toString());
+            map.put(template.replace(".vm", ""), sw.toString());
             System.out.println(sw);
         }