fixing snapshot testss