diff --git a/src/main/java/com/zhangmeng/tools/controller/MySQLCodeGenController.java b/src/main/java/com/zhangmeng/tools/controller/MySQLCodeGenController.java
index 2d46008..9536fff 100644
--- a/src/main/java/com/zhangmeng/tools/controller/MySQLCodeGenController.java
+++ b/src/main/java/com/zhangmeng/tools/controller/MySQLCodeGenController.java
@@ -261,14 +261,14 @@ public class MySQLCodeGenController {
             AlertUtils.alert_warning("请预览之后在进行操作!");
             return;
         }
-
+//        assignment_to_filed(generateDetail);
         GenerateInput generateInput = new GenerateInput();
         generateInput.setTableName(generateDetail.getTableName());
         generateInput.setPath(path_field.getText());
 
         generateInput.setControllerName(controller_class_name_field.getText());
         generateInput.setDaoName(dao_clas_name_field.getText());
-        generateInput.setBeanName(generateDetail.getBeanName());
+        generateInput.setBeanName(bean_class_name_field.getText());
 
         generateInput.setDaoPackageName(dao_package_name_field.getText());
         generateInput.setControllerPkgName(controller_package_name_field.getText());