Correct test-case to match current functionality