Added missing import.

Missing import breaks current monkey version. Accidentally removed import in previous changes.