Use future import to avoid branching on Python version