Fix issue 4: missing register button