Fix assertion in tests
Merge pull request #2 from just-boris/master

Fix assertion in tests