Fix for updating connection