fix python code typo