Fix registration :)