Correct import syntax so it works in both 2.6+ and 3+