Fixes disconnect