Fixes failing build by adding needed imports. bjmcmorran