Re-add missing import