Updated test to use new index