Fix missing ast done call