From d1497453baedea20905ab85265641a255177e224 Mon Sep 17 00:00:00 2001
From: Tangem <git@tangem.com>
Date: Fri, 3 Jul 2026 14:28:33 +0300
Subject: [PATCH] Updated on 2026-08-14

---
 .../kotlin/com/tangem/tests/referral/ReferralTest.kt      | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/app/src/androidTest/kotlin/com/tangem/tests/referral/ReferralTest.kt b/app/src/androidTest/kotlin/com/tangem/tests/referral/ReferralTest.kt
index 93ca99d3ef..f0d93d7ea1 100644
--- a/app/src/androidTest/kotlin/com/tangem/tests/referral/ReferralTest.kt
+++ b/app/src/androidTest/kotlin/com/tangem/tests/referral/ReferralTest.kt
@@ -56,8 +56,11 @@ class ReferralTest : BaseTestCase() {
             }
             step("Return to the 'Main' screen") {
                 tapBackButton()
+                onWalletSettingsScreen { screenContainer.assertIsDisplayed() }
                 tapBackButton()
+                onDetailsScreen { screenContainer.assertIsDisplayed() }
                 tapBackButton()
+                onMainScreen { screenContainer.assertIsDisplayed() }
             }
             step("Verify network $tokenNetwork and token $token is displayed on 'Main' screen") {
                 onMainScreen {
@@ -102,10 +105,13 @@ class ReferralTest : BaseTestCase() {
             }
             step("Return to the 'Main' screen") {
                 tapBackButton()
+                onWalletSettingsScreen { screenContainer.assertIsDisplayed() }
                 tapBackButton()
+                onDetailsScreen { screenContainer.assertIsDisplayed() }
                 tapBackButton()
+                onMainScreen { screenContainer.assertIsDisplayed() }
             }
-            step("Verify $token is displaying on 'Main' screen") {
+            step("Verify $token is displayed on 'Main' screen") {
                 onMainScreen { tokenWithTitleAndAddress(token) }
             }
         }