diff --git a/android/quest/src/androidTest/java/org/smartregister/fhircore/quest/integration/ui/register/RegisterScreenTest.kt b/android/quest/src/androidTest/java/org/smartregister/fhircore/quest/integration/ui/register/RegisterScreenTest.kt
index af2bd8542c..4f06616a7f 100644
--- a/android/quest/src/androidTest/java/org/smartregister/fhircore/quest/integration/ui/register/RegisterScreenTest.kt
+++ b/android/quest/src/androidTest/java/org/smartregister/fhircore/quest/integration/ui/register/RegisterScreenTest.kt
@@ -182,9 +182,6 @@ class RegisterScreenTest {
                 ),
             ),
         registerId = "register101",
-        totalRecordsCount = 1,
-        filteredRecordsCount = 0,
-        pagesCount = 0,
         progressPercentage = flowOf(0),
         isSyncUpload = flowOf(false),
         params = emptyList(),
@@ -207,6 +204,12 @@ class RegisterScreenTest {
         pagingItems = pagingItems,
         navController = rememberNavController(),
         decodeImage = null,
+        registerUiCountState =
+          RegisterUiCountState(
+            totalRecordsCount = 1,
+            filteredRecordsCount = 0,
+            pagesCount = 0,
+          ),
       )
     }