Fix a small input bug