Fix: Add missing import