Revert "Remove superfluous import"

This reverts commit ebf1615283effa001742b709d7f7b4ece0ed67cc.