diff --git a/datacontract/data_contract.py b/datacontract/data_contract.py index 491fb5ee..8b478e32 100644 --- a/datacontract/data_contract.py +++ b/datacontract/data_contract.py @@ -208,6 +208,8 @@ def test(self) -> Run: if self._examples: server_name = "examples" server = self._get_examples_server(data_contract, run, tmp_dir) + if self._server: + server = data_contract.servers.get(self._server) else: server_name = list(data_contract.servers.keys())[0] server = data_contract.servers.get(server_name)