Working on fixing testSnapshot