Fix broken call to register