Adding missing input and correcting a type