Minor mistake fix