Fixing error

Fixing the variable refrences