Correcting entity merging in AI search systems