Fix missing import

This got broken during the latest large merge.