Updated unit test to reflect change in key dimension and content format