--upgrade fix