USB disconnect fix