Fix tests: