Add missing import from future