add missed use statement, and remove another

This use I missed from the last commit...