Correct stupid coding mistake