Correcting a mistake from previous commit