Fixed missing use statement and removed unused one.