Fix registration thing