Add lazy initialization to the register machine