Updated test to reflect the returned data