Fixed missing import

Fixed a missing import from 11c9694