fixed the seg fault