Fix incorrect args