upgrade bug fix