Update error type for the latest implementation