Fixed a minor error with a variable