seg fault bug fix