Corrected small mistake