fix undefined variable from last commit