FIX: Only import when necessary