fixed seg fault prob