Fix import in 2to3 transition