[Registration] Fix misplaced registration