Fix tests after the parse change