Fixed a bug for a missing use statement