upgrade bug fixed