fix connection class