Register button fix