updating unit tests to match new data structure