Updated UndefinedVariables for the new AST.

Signed-off-by: Yorick Peterse <yorickpeterse@gmail.com>