One more missing import

Change-Id: I404b27ae96b534551784dc3942fb7267469ebabb
Signed-off-by: Evan Anderson <evan1124@gmail.com>