Client disconnect fix.