Fix a input issue #2