[noticket] Add missing import - probably arose due to cherry-pick.