Fixing seg fault.