Revert "remove unused import"

This reverts commit bdef39c061a3bb0ddfa434bc7397f3420b12a382.