Fix part 1

Fix part 1