Fixed a seg fault