Connection problem fix