Fix AfterStatement parsing