diff --git a/app/src/test/kotlin/com/vrem/wifianalyzer/wifi/band/WiFiChannelCountryTest.kt b/app/src/test/kotlin/com/vrem/wifianalyzer/wifi/band/WiFiChannelCountryTest.kt
index 42272a22..cf4e3e40 100644
--- a/app/src/test/kotlin/com/vrem/wifianalyzer/wifi/band/WiFiChannelCountryTest.kt
+++ b/app/src/test/kotlin/com/vrem/wifianalyzer/wifi/band/WiFiChannelCountryTest.kt
@@ -33,7 +33,7 @@ class WiFiChannelCountryTest {
         val expected = Locale.US
         val actual = WiFiChannelCountry.find(expected.country)
         assertThat(actual.countryCode).isEqualTo(expected.country)
-        assertThat(actual.countryName(expected)).isEqualTo(expected.displayCountry)
+        assertThat(actual.countryName(expected)).isEqualTo(expected.getDisplayCountry(expected))
     }
 
     @Test
diff --git a/app/src/test/kotlin/com/vrem/wifianalyzer/wifi/band/WiFiChannelsParameterizedTest.kt b/app/src/test/kotlin/com/vrem/wifianalyzer/wifi/band/WiFiChannelsParameterizedTest.kt
index 0447d2bc..94e3124a 100644
--- a/app/src/test/kotlin/com/vrem/wifianalyzer/wifi/band/WiFiChannelsParameterizedTest.kt
+++ b/app/src/test/kotlin/com/vrem/wifianalyzer/wifi/band/WiFiChannelsParameterizedTest.kt
@@ -20,11 +20,11 @@ package com.vrem.wifianalyzer.wifi.band
 import com.vrem.util.EMPTY
 import com.vrem.wifianalyzer.wifi.model.WiFiWidth
 import org.assertj.core.api.Assertions.assertThat
+import org.junit.Test
 import org.junit.runner.RunWith
 import org.junit.runners.Parameterized
 import org.junit.runners.Parameterized.Parameter
 import org.junit.runners.Parameterized.Parameters
-import kotlin.test.Test
 
 @RunWith(Parameterized::class)
 class WiFiChannelsParameterizedTest {
@@ -166,7 +166,7 @@ class WiFiChannelsParameterizedTest {
     @Test
     fun availableChannelsUsingWiFiBandAndCountry() {
         countries.forEach { country ->
-            assertThat(fixture.availableChannels(wiFiBand, country.countryCode).map { it -> it.channel })
+            assertThat(fixture.availableChannels(wiFiBand, country.countryCode).map { it.channel })
                 .describedAs("Country: ${country.countryCode}")
                 .containsExactlyElementsOf(country.channels(wiFiBand))
         }