Update test to new dict format