fix hw4 bug