fixed tests for new connection api