Revert "Remove unused import"

This reverts commit 046e54bc46f02c1aff72506eb0e470f55b67a180.