Fix minor variable error