Fix tests around ES suggest API