Fix identity tests