Tests: Fixed test to match new story array