Fixed the register thing