a little change in register usage and fix a mistake