HW2-3 fix test faliure