Fix import on py27