Fixed missing use statements