Minor fix, I'm a bit embarassed