TST: fixed test to support change