Fix missing inclusion