Try to fix auto restart feature on upgrade