Fixed a seg fault
Fixed a seg fault