Fix an ambiguity in puzzle 4