fix the error in pretest 3 & 4