Fix correct argument after updates