Refactored failing test to reflect current logic