Fix optimize bug