[TASK] Fixup the annotation for