Add TOS checking to register