So I should just remove the register_opt() decorator or, do I need to add your call after for the optimizer_including trick to work?