Ok, entity prediction works now.