Fixed a small bug in register