Update utils ast functions
Reflect changes in utils ast functions in tests