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