Update for ast changes

Fixes #65