Changed this test case to use take, which I think is a better way to verify this structure