Oops, fixed undefined var reference