Fix test import for py26 compat