Fix hot reboot

Change-Id: Ic9973c8d2bfc055608c074b917e71d108016c4e4