Fix too-many-arguments error