Add the required import.

I'm not sure if this is the right way to do this, but I tried to
use this chunk of the documentation without reading the rest and
was unaware this was needed, thought it might help the next person
trying to cut and paste away.