Merge pull request #2 from just-boris/master

Fix assertion in tests