Updated test to match spec