Fixed the connection bug!