Fixed seg fault