Fix solution for riddle #7