Re-adding set class to ast so naming can be retained.