Another failing test fix