Fix arg signature