fixed a small bug in the optimization step and the method finally can work