diff --git a/test/integration/api.bats b/test/integration/api.bats
index a6a233a520..8146e8a2fc 100644
--- a/test/integration/api.bats
+++ b/test/integration/api.bats
@@ -55,15 +55,11 @@ function teardown() {
 	# jimmyxian/centos7-wssh is an image with websocket CLI(WSSH) wirtten in Nodejs
 	# if connected successfull, it returns two lines, "Session Open" and "Session Closed"
 	# Note: with stdout=1&stdin=1&stream=1: it can be used as SSH
-	docker_host pull jimmyxian/centos7-wssh
-	[ "$status" -eq 0 ]
-
 	URL="ws://${SWARM_HOST}/${CLIENT_API_VERSION}/containers/test_container/attach/ws?stderr=1"
 	run docker_host run --rm --net=host jimmyxian/centos7-wssh wssh $URL
 	[ "$status" -eq 0 ]
-	[ "${#lines[@]}" -eq 2 ]
-	[[ "${lines[0]}" == *"Session Open"* ]]
-	[[ "${lines[1]}" == *"Session Closed"* ]]
+	[[ "${output}" == *"Session Open"* ]]
+	[[ "${output}" == *"Session Closed"* ]]
 }
 
 @test "docker build" {