Correct the segmentation fault!! ^_^