diff --git a/VOL.Ai/Services/AiAppService.cs b/VOL.Ai/Services/AiAppService.cs
index a5286df..02e560c 100644
--- a/VOL.Ai/Services/AiAppService.cs
+++ b/VOL.Ai/Services/AiAppService.cs
@@ -1032,7 +1032,7 @@ namespace VOL.Ai.Services
             var heartRateDatas = _mapper.Map<List<Ai_HeartRateData>>(paramDto.Datas);
 
             var classIds = paramDto.Datas.Select(x => x.ClassId).Distinct().ToList();
-            var classList = await _classRepository.FindAsync(x => x.SchoolCode == paramDto.SchoolCode && classIds.Contains(x.Id));
+            var classList = await _classRepository.DbContext.Set<JS_Class>().Where(x => x.SchoolCode == paramDto.SchoolCode && classIds.Contains((int)x.Id)).ToListAsync();
             var classDict = classList.ToDictionary(x => x.Id, x => x);
 
             var timeNow = DateTime.Now;
@@ -1046,7 +1046,7 @@ namespace VOL.Ai.Services
                     heartRateData.ClassRoomRecordId = paramDto.ClassRoomRecordId;
                     heartRateData.ClassroomStageId = paramDto.ClassroomStageId;
                     heartRateData.TeacherId = paramDto.TeacherId;
-                    heartRateData.GradeId = classInfo.GradeId;
+                    heartRateData.GradeId = (int)classInfo.GradeId;
                     heartRateData.GradeName = classInfo.GradeName;
                     heartRateData.ClassName = classInfo.ClassName;
                     heartRateData.ScoreTime = timeNow;