add missing import

got removed during bad merge