fixed a seg fault bug