Fix tests
release 0.17.2