Fix stupid missing variable error. Doh