Add missing includes
Fix compilation?