Fix for managing disconection from server side.