small change to connection class