fix unit tests
Merge pull request #6 from metridoc/current-progress

fix unit tests