Fix test for change to Identity