Fixed snapshot tests