Fix: Added missing includes