Fix unit tests
Fix
release 0.5.6