fix snapshot tests