Fix missing import.

Reported by Zero Cho.

Reviewed at https://rbcommons.com/s/twitter/r/1483/