Correct mistake from previous commit