Fix __future__ import.