Fixed seg fault bug