Fix .py template