diff --git a/src/test/resources/generator/template/controller.ftl b/src/test/resources/generator/template/controller.ftl
index d14c445..45e6583 100644
--- a/src/test/resources/generator/template/controller.ftl
+++ b/src/test/resources/generator/template/controller.ftl
@@ -48,7 +48,7 @@ public class ${modelNameUpperCamel}Controller {
     @PostMapping("/list")
     public Result list(Integer page, Integer size) {
         PageHelper.startPage(page, size);
-        List<User> list = userService.findAll();
+        List<${modelNameUpperCamel}> list = ${modelNameLowerCamel}Service.findAll();
         PageInfo pageInfo = new PageInfo(list);
         return ResultGenerator.genSuccessResult(pageInfo);
     }