Fix bug with register
Now do fix that bug