Fix tests
release 0.1.8