Support 2.6 with conditional unittest import