Minor fix to Python import syntax