FIXED mising variable

Readded line that was deleted in the last commit