Fix our test
Merge pull request #1 from zckrs/fix-test

Fix test