Fixed the seg fault.