Fix tests and bump the version