diff --git a/src/test/resources/generator/template/controller.ftl b/src/test/resources/generator/template/controller.ftl
index 530190d..4a1f0bb 100644
--- a/src/test/resources/generator/template/controller.ftl
+++ b/src/test/resources/generator/template/controller.ftl
@@ -3,6 +3,8 @@ import ${basePackage}.core.Result;
 import ${basePackage}.core.ResultGenerator;
 import ${basePackage}.model.${modelNameUpperCamel};
 import ${basePackage}.service.${modelNameUpperCamel}Service;
+import com.github.pagehelper.PageHelper;
+import com.github.pagehelper.PageInfo;
 import org.springframework.web.bind.annotation.PostMapping;
 import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.RestController;