Fixed missing use statements

--