Adding an import whose absence was preventing the example from compiling