Small var bug fix