Correction seg fault