Bug fix in python code