[TASK] Add missing argument registration