diff --git a/src/main/java/org/yangliu/codegenerate/Main.java b/src/main/java/org/yangliu/codegenerate/Main.java
index 92fc6dc..4240392 100644
--- a/src/main/java/org/yangliu/codegenerate/Main.java
+++ b/src/main/java/org/yangliu/codegenerate/Main.java
@@ -118,7 +118,7 @@ public class Main {
         //生成Mapper.xml
         String mapperXmlParsePath = project_resource_mapper_path;
         MarkerGenerate.generate(templatepath, bussi_mapperxml_template, mapperXmlParsePath, entityName + "Mapper.xml", dataMap);
-        /*//生成Service
+        //生成Service
         String serviceParsePath = parsePath + "service" + File.separator;
         MarkerGenerate.generate(templatepath,bussi_service_template,serviceParsePath,String.format("I%sService.java",entityName),dataMap);
         //生成ServiceImpl
@@ -126,7 +126,7 @@ public class Main {
         MarkerGenerate.generate(templatepath,bussi_serviceimpl_template,serviceImplParsePath,String.format("%sServiceImpl.java",entityName),dataMap);
         //生成Controller
         String controllerParsePath = parsePath + "controller" + File.separator;
-        MarkerGenerate.generate(templatepath,bussi_controller_template,controllerParsePath,String.format("%sController.java",entityName),dataMap);*/
+        MarkerGenerate.generate(templatepath,bussi_controller_template,controllerParsePath,String.format("%sController.java",entityName),dataMap);
 
     }
 
@@ -178,5 +178,4 @@ public class Main {
         return tableEntityVoList;
     }
 
-
 }
diff --git a/src/main/resources/yangliu/XXXController.javai b/src/main/resources/yangliu/XXXController.javai
index 0de31ad..7ef3db6 100644
--- a/src/main/resources/yangliu/XXXController.javai
+++ b/src/main/resources/yangliu/XXXController.javai
@@ -6,8 +6,8 @@ import com.enjoy.common.core.web.page.TableDataInfo;
 import com.enjoy.common.log.annotation.Log;
 import com.enjoy.common.log.enums.BusinessType;
 import ${bussiPackage}.${entityPackage}.${entityName};
-import ${bussiPackage}.dto.${entityName}Dto;
-import ${bussiPackage}.vo.${entityName}Vo;
+import ${bussiPackage}.dto.${entityName}DTO;
+import ${bussiPackage}.vo.${entityName}VO;
 import ${bussiPackage}.service.I${entityName}Service;