fix snapshot tests
Merge pull request #7 from mistifyio/tests

fix snapshot tests