Fixes test for collection