Revert "Remove useless import"

This reverts commit 73a2f0cf499f0201c8298eadf5999c67c2b31696.