Fix snapshot tests