add missing import

Somehow this was missed