Fix tests.
release 0.17.1