Replace setCountry with setCustomRelay

2 files changed, 4 insertions, 4 deletions

diff --git a/test/connect.spec.js b/test/connect.spec.js
index 9ecedb53a1..98aba67356 100644
--- a/test/connect.spec.js
+++ b/test/connect.spec.js
@@ -7,13 +7,13 @@ import { IpcChain } from './helpers/IpcChain';
 
 describe('connect', () => {
 
-  it('should invoke set_country and then connect in the backend', (done) => {
+  it('should invoke set_custom_relay and then connect in the backend', (done) => {
     const { store, mockIpc, backend } = setupBackendAndStore();
 
     const chain = new IpcChain(mockIpc);
-    chain.require('setCountry')
+    chain.require('setCustomRelay')
       .withInputValidation(
-        (country) => expect(country).to.equal('example.com')
+        (relayHostName) => expect(relayHostName).to.equal('example.com')
       )
       .done();
 
diff --git a/test/mocks/ipc.js b/test/mocks/ipc.js
index d28c49e6f6..472811ef4c 100644
--- a/test/mocks/ipc.js
+++ b/test/mocks/ipc.js
@@ -27,7 +27,7 @@ export function newMockIpc() {
     setAccount: () => {
       return new Promise(r => r());
     },
-    setCountry: () => {
+    setCustomRelay: () => {
       return new Promise(r => r());
     },
     connect: () => {