add code in .py file to avoid error