[BUGFIX] Add missing registration of argument