Updated collapse unit test