Fixed problem with breaking connection