Update test to latest changes.