Fix the variable name I forgot to change
Fix some whitespace as well