Fixed minor upgrade error
Merged