update tests - TODO: fix collection tests