Fix error in Connection class (maybe)