Added missing import (#11982)

Change-Id: Ie647ac58f028abd96ddaaf2952b606069071500e