fixing part 2

fixing part 2