made change in data structure more clear in test