Make __future__ import check more robust