Fixing seg fault