fix lost upgrade highlight