Fix missing import for MODE