Fixed up reconnect logic (had broken it with earlier change)