fix bugs in unittest
2 files changed