fix hw5.2