Update unit test now that code_error is gone