diff --git a/src/hooks/useRegisterImage.ts b/src/hooks/useRegisterImage.ts
index c6f028e..73b1299 100644
--- a/src/hooks/useRegisterImage.ts
+++ b/src/hooks/useRegisterImage.ts
@@ -19,16 +19,16 @@ export default function useRegisterImage(canPreview: boolean, data: ImageElement
   // Keep order start
   // Resolve https://github.com/ant-design/ant-design/issues/28881
   // Only need unRegister when component unMount
-  // React.useEffect(() => {
-  //   if (groupContext) {
-  //     return groupContext.register(id, registerData);
-  //   }
-  // }, []);
-
   React.useEffect(() => {
     if (groupContext) {
       return groupContext.register(id, registerData);
     }
+  }, []);
+
+  React.useEffect(() => {
+    if (groupContext) {
+      groupContext.register(id, registerData);
+    }
   }, [canPreview, data]);
 
   return id;