fix missing import on prev commit

Forgot to wait for travis.