Fix tests
Version 0.0.6 release