Fixed seg fault error