Fixing mistake from previous commit