Fix the test, we should be looking at the second element instead