Fix snapshot generation