Fixed the seg fault