Fix broken registration