Fix an undefined variable error, see #7