Fixed seg fault
Fixed seg fault