Fix test compile error
Merge pull request #11 from alexeyvo/fix_test

Fix test compile error