Update compliance test to account for API changes.