Fix snapshot unit tests.