diff --git a/test/integration/api.bats b/test/integration/api.bats
index a9fca844d5..b0d9b0a079 100644
--- a/test/integration/api.bats
+++ b/test/integration/api.bats
@@ -54,7 +54,7 @@ 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 pull jimmyxian/centos7-wssh
+	docker_host pull jimmyxian/centos7-wssh
 
 	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