Added missing import.

I think your last merge was missing an import.