Fix: Added updateParentState expectations in tests