Fix tests in error