Fixed unit test and updated snapshot