diff --git a/src/test/resources/generator/template/controller.ftl b/src/test/resources/generator/template/controller.ftl
index 4a1f0bb..d14c445 100644
--- a/src/test/resources/generator/template/controller.ftl
+++ b/src/test/resources/generator/template/controller.ftl
@@ -10,6 +10,7 @@ import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.RestController;
 
 import javax.annotation.Resource;
+import java.util.List;
 
 /**
 * Created by ${author} on ${date}.