fix missing import

@gwprice