diff --git a/src/bitmessagekivy/tests/test_create_random_address.py b/src/bitmessagekivy/tests/test_create_random_address.py
index 87698369..32ec87a4 100644
--- a/src/bitmessagekivy/tests/test_create_random_address.py
+++ b/src/bitmessagekivy/tests/test_create_random_address.py
@@ -29,18 +29,15 @@ class CreateRandomAddress(TeleniumTestProcess):
         )
         # Assert the checkbox is rendered
         self.assertExists(
-            '//Login//Screen[@name=\"check_screen\"]//AnchorLayout[1]/Check', timeout=5)
-        # Checking Status of checkbox before click
-        self.assertEqual(self.cli.getattr(
-            '//Login//Screen[@name=\"check_screen\"]//AnchorLayout[1]/Check[@active]', 'active'), False
+            '//Login//Screen[@name=\"check_screen\"]//AnchorLayout[1]/Check[@active=false]', timeout=5
         )
         # Clicking on the checkbox
         self.cli.wait_click(
             '//Login//Screen[@name=\"check_screen\"]//AnchorLayout[1]/Check', timeout=5
         )
         # Checking Status of checkbox after click
-        self.assertEqual(self.cli.getattr(
-            '//Login//Screen[@name=\"check_screen\"]//AnchorLayout[1]/Check', 'active'), True
+        self.assertExists(
+            '//Login//Screen[@name=\"check_screen\"]//AnchorLayout[1]/Check[@active=true]', timeout=5
         )
         # Checking the Proceed Next button is rendered or not
         self.assertExists(
diff --git a/src/bitmessagekivy/tests/test_network_screen.py b/src/bitmessagekivy/tests/test_network_screen.py
index f307dfae..cc2eb727 100644
--- a/src/bitmessagekivy/tests/test_network_screen.py
+++ b/src/bitmessagekivy/tests/test_network_screen.py
@@ -12,7 +12,7 @@ class NetworkStatusScreen(TeleniumTestProcess):
     def test_network_status(self):
         """Show NetworkStatus"""
         # This is for checking Current screen
-        self.assert_wait_no_except('//ScreenManager[@current]', timeout=15, value='login')
+        self.assert_wait_no_except('//ScreenManager[@current]', timeout=15, value='inbox')
         # Method to open side navbar
         # due to rapid transition effect, it doesn't click on menu-bar
         self.open_side_navbar()