Fixed device restart handling during firmware update.