fix part 2