Revert "Compile error: missing import"

This reverts commit de3840d8f4d19ad9a9072e83a9f68ee6c78f4328.