Fix import from future