fix problem 2