updated unit test to check for lazy collection