Fix should failures in tests.