Fix missing import & broken build.

That's what I get for editing through Github's web UI...