Fix identity unit test