Fixed python code error