Fixed missing use statement