Fixed accidental oversight with last commit