Fixed another small mistake