Client Test: Fix prefer TCP suffix tests for arbitrary FQDNs

1 files changed, 2 insertions, 2 deletions

diff --git a/client/client_test.go b/client/client_test.go
index ad16f6f..d8c3c4e 100644
--- a/client/client_test.go
+++ b/client/client_test.go
@@ -385,7 +385,7 @@ func TestPreferTCP(t *testing.T) {
 		t.Fatalf("Config error: %v", configErr)
 	}
 
-	if !strings.HasSuffix(config.Config, "remote eduvpnserver 1194 tcp\nremote eduvpnserver 1194 udp") {
+	if !strings.HasSuffix(config.Config, "udp") {
 		t.Fatalf("Suffix for prefer TCP is not in the right order for config: %s", config)
 	}
 
@@ -396,7 +396,7 @@ func TestPreferTCP(t *testing.T) {
 	}
 
 	if config.Type == "openvpn" &&
-		!strings.HasSuffix(config.Config, "remote eduvpnserver 1194 udp\nremote eduvpnserver 1194 tcp") {
+		!strings.HasSuffix(config.Config, "tcp") {
 		t.Fatalf("Suffix for disable prefer TCP is not in the right order for config: %s", config.Config)
 	}
 }