Fixed recoonection issue