diff --git a/cli/server_test.go b/cli/server_test.go
index 1532793caa321..cad67f3d34b16 100644
--- a/cli/server_test.go
+++ b/cli/server_test.go
@@ -267,6 +267,8 @@ func TestServer(t *testing.T) {
 
 		<-deployment
 		<-snapshot
+		cancelFunc()
+		<-errC
 	})
 }