Fix more missing variables

I can change it to something else if you want me to :)