Fix my last fix

Forgot to add this line. :/