Fix unittest failure
Merge pull request #2 from sinkuu/fix_unittest

Fix unittest failure