diff --git a/tests/integration/api_swarm_test.py b/tests/integration/api_swarm_test.py
index d06cac21bd..0a2f69f3c9 100644
--- a/tests/integration/api_swarm_test.py
+++ b/tests/integration/api_swarm_test.py
@@ -168,7 +168,7 @@ def test_remove_main_node(self):
         with pytest.raises(docker.errors.APIError) as e:
             self.client.remove_node(node_id)
 
-        assert e.value.response.status_code == 500
+        assert e.value.response.status_code >= 400
 
         with pytest.raises(docker.errors.APIError) as e:
             self.client.remove_node(node_id, True)