Revert "Remove unused import"

This reverts commit 040712c6fdd725c08bd0e106e2b987ec6e7a8780.