Small fix to direct entity searching