Fixed connection problem

Fixed connection problem