Fix comment problem for python