Corrected __future__ import in test.py