Fixed some AST shieeetytttt