Fix test based on change to error code