[TASK] Correct argument registration