Fix failed test
Merge pull request #63 from alexeyvo/v2

Fix failed test