Add missing use statement

Change-Id: I33b2170858e6e4f36ef237194da9736c0067181a