diff --git a/api/problem.go b/api/problem.go
index faa977a..a973aba 100644
--- a/api/problem.go
+++ b/api/problem.go
@@ -5,7 +5,6 @@ import (
 	"online_code/core"
 	"online_code/models/param"
 	"online_code/repository"
-	"online_code/utils"
 )
 
 type problemApi struct{}
@@ -26,8 +25,7 @@ func (problemApi) GetProbleList(ctx *gin.Context) {
 		core.R(ctx).FailWithMessage("获取题目列表失败: " + err.Error())
 		return
 	}
-	// 计算总页码
-	totalPage := utils.GenTotalPage(count, p.Size)
+
 	// 返回结果
-	core.R(ctx).OkWithData(core.PageData{Current: p.Current, Size: p.Size, Total: count, TotalPage: totalPage, Records: records})
+	core.R(ctx).OkDataPage(count, core.PageData{Current: p.Current, Size: p.Size, Total: count, Records: records})
 }
diff --git a/repository/problem.go b/repository/problem.go
index e02fa0c..e63ea29 100644
--- a/repository/problem.go
+++ b/repository/problem.go
@@ -13,7 +13,7 @@ func ProblemService() *problemService {
 }
 
 // GetList 获取题目列表
-func (problemService) GetList(p param.GetProblemList) (records entity.ProblemBasic, count int64, err error) {
+func (problemService) GetList(p param.GetProblemList) (records []entity.ProblemBasic, count int64, err error) {
 	sel := client.MySQL.Scopes(page(p.Current, p.Size)).Preload("ProblemCategories").Preload("ProblemCategories.CategoryBasic")
 
 	if p.Keyword != "" {