fixing broken import

Somehow managed to miss an add.