Undoing my last commit. I misread the code before.