Add the test for collection