Updated test to include connect