Accidentally removed a line of code, it's back now