diff --git a/backend-java/src/test/java/com/nanri/aiimage/modules/collectdata/service/CollectDataNoUploadStaleTest.java b/backend-java/src/test/java/com/nanri/aiimage/modules/collectdata/service/CollectDataNoUploadStaleTest.java
index 6c3c2db7..601de0b9 100644
--- a/backend-java/src/test/java/com/nanri/aiimage/modules/collectdata/service/CollectDataNoUploadStaleTest.java
+++ b/backend-java/src/test/java/com/nanri/aiimage/modules/collectdata/service/CollectDataNoUploadStaleTest.java
@@ -85,7 +85,7 @@ class CollectDataNoUploadStaleTest {
         invokeNoUpload();
 
         verify(taskDistributedLockService, never()).acquire(any(), anyLong(), anyLong());
-        verify(fileTaskMapper, never()).updateById(any());
+        verify(fileTaskMapper, never()).updateById(any(FileTaskEntity.class));
     }
 
     @Test
@@ -96,7 +96,7 @@ class CollectDataNoUploadStaleTest {
         invokeNoUpload();
 
         verify(taskDistributedLockService, never()).acquire(any(), anyLong(), anyLong());
-        verify(fileTaskMapper, never()).updateById(any());
+        verify(fileTaskMapper, never()).updateById(any(FileTaskEntity.class));
     }
 
     @Test
@@ -117,7 +117,7 @@ class CollectDataNoUploadStaleTest {
 
         invokeNoUpload();
 
-        verify(fileTaskMapper, never()).updateById(any());
+        verify(fileTaskMapper, never()).updateById(any(FileTaskEntity.class));
     }
 
     @Test