Corrected the wrong code