diff --git a/tests/test_pineap_proxy.py b/tests/test_pineap_proxy.py
index ee39b72..614ac5d 100644
--- a/tests/test_pineap_proxy.py
+++ b/tests/test_pineap_proxy.py
@@ -129,7 +129,7 @@ class PineapProxyTest(unittest.TestCase):
         status, payload = server.h_pineap_wifi_get_ap(ctx())
         self.assertEqual(status, 200)
         self.assertEqual(payload['wpa'], {'ssid': 'Evil1', 'passphrase': 'sekret', 'enctype': 'psk2',
-                                          'hidden': False, 'enabled': True})
+                                          'hidden': False, 'enabled': True, 'channel': 6})
         self.assertEqual(payload['open']['enabled'], False)
         self.assertEqual(payload['open']['ssid'], 'pager-open')
         self.assertEqual(payload['open']['bssid'], 'DE:AD:BE:EF:00:01')
@@ -150,7 +150,7 @@ class PineapProxyTest(unittest.TestCase):
 
         def fake_run(args):
             run_calls.append(args)
-            if args[0] == 'uci' and args[1] == 'show':
+            if args[0] == 'uci' and args[1] == 'show' and args[2] == 'wireless.radio0':
                 return 0, "wireless.radio0.channel='1'\n", ''
             return 0, '', ''