updated unit tests for multiple ids