Stupid mistake.

I forgot to change one line.