diff --git a/WooCommerce/src/test/kotlin/com/woocommerce/android/ui/woopos/home/totals/WooPosTotalsViewModelTest.kt b/WooCommerce/src/test/kotlin/com/woocommerce/android/ui/woopos/home/totals/WooPosTotalsViewModelTest.kt
index 28131673e75..ebb6d9674b1 100644
--- a/WooCommerce/src/test/kotlin/com/woocommerce/android/ui/woopos/home/totals/WooPosTotalsViewModelTest.kt
+++ b/WooCommerce/src/test/kotlin/com/woocommerce/android/ui/woopos/home/totals/WooPosTotalsViewModelTest.kt
@@ -1194,8 +1194,8 @@ class WooPosTotalsViewModelTest {
             )
 
             // THEN
-            val totals = viewModel.state.value as WooPosTotalsViewState.Totals
-            assertTrue(totals.isFreeOrder)
+            val checkout = viewModel.state.value as WooPosTotalsViewState.Checkout
+            assertTrue(checkout.isFreeOrder)
         }
 
     @Test
@@ -1253,8 +1253,8 @@ class WooPosTotalsViewModelTest {
             )
 
             // THEN
-            val totals = viewModel.state.value as WooPosTotalsViewState.Totals
-            assertFalse(totals.isFreeOrder)
+            val checkout = viewModel.state.value as WooPosTotalsViewState.Checkout
+            assertFalse(checkout.isFreeOrder)
         }
 
     private fun createNonEmptyOrder() = Order.getEmptyOrder(