Set the collectionClass like a boss