Fixed another import.

Fixed another import statement I missed.