[TASK] Fix annotation for validation