diff --git a/packages/chicken/test/integration/steward_workflow_integration_test.dart b/packages/chicken/test/integration/steward_workflow_integration_test.dart
index 58025c8..1749442 100644
--- a/packages/chicken/test/integration/steward_workflow_integration_test.dart
+++ b/packages/chicken/test/integration/steward_workflow_integration_test.dart
@@ -138,6 +138,7 @@ void main() {
         when(
           () => mockRemote.getInventory(
             token: token,
+            role: 'steward',
             cancelToken: any(named: 'cancelToken'),
           ),
         ).thenAnswer((_) async => expectedInventory);
@@ -147,7 +148,10 @@ void main() {
         ).thenAnswer((_) async => expectedKillHouseInfo);
 
         // Act - Step 1: Get inventory
-        final inventory = await chickenRepository.getInventory(token: token);
+        final inventory = await chickenRepository.getInventory(
+          token: token,
+          role: 'steward',
+        );
 
         // Act - Step 2: Get kill house distribution info
         final killHouseInfo = await chickenRepository
@@ -160,6 +164,7 @@ void main() {
         verify(
           () => mockRemote.getInventory(
             token: token,
+            role: 'steward',
             cancelToken: any(named: 'cancelToken'),
           ),
         ).called(1);
@@ -404,18 +409,23 @@ void main() {
         when(
           () => mockRemote.getInventory(
             token: token,
+            role: 'steward',
             cancelToken: any(named: 'cancelToken'),
           ),
         ).thenAnswer((_) async => null);
 
         // Act
-        final inventory = await chickenRepository.getInventory(token: token);
+        final inventory = await chickenRepository.getInventory(
+          token: token,
+          role: 'steward',
+        );
 
         // Assert
         expect(inventory, isNull);
         verify(
           () => mockRemote.getInventory(
             token: token,
+            role: 'steward',
             cancelToken: any(named: 'cancelToken'),
           ),
         ).called(1);