Update to match latest error API