Change connect function slightly.