fix bug. Segment fault.