Revert "remove unused import"

This reverts commit a645d5a71c6c956f1195f91bd9be09e46e7a1d90.