Correction piece of code put on wrong place