diff --git a/eatery-service/src/test/java/com/drinkool/services/ShiftServiceTest.java b/eatery-service/src/test/java/com/drinkool/services/ShiftServiceTest.java
index 854b33c..88f4033 100644
--- a/eatery-service/src/test/java/com/drinkool/services/ShiftServiceTest.java
+++ b/eatery-service/src/test/java/com/drinkool/services/ShiftServiceTest.java
@@ -102,7 +102,9 @@ public class ShiftServiceTest {
 
 		given()
 			.contentType("application/json")
-			.header(InternalValue.userId, staffId.toString()) // Mock user ID
+			.header(InternalValue.role, Role.Staff)
+			.header(InternalValue.userId, staffId.toString()) 
+			.header(InternalValue.managerId, ownerId.toString())
 			.body("{ \"query\": \"" + mutation + "\" }")
 			.when()
 			.post("/graphql")