Address more review comments.
Fix bug in InputSingleRegister().