Fix tests

OK