some updates for new haytack API