Include forgotten imports.

This fixes the build.

Change-Id: Id132f4342adb9ed20116191086f157ca7f7cf515