fix issue in case of reboot