add missing import

Not sure how this snuck in under the radar