Fix code to compile: Part 2