Fix make_input.py