quick fix for part 2