Fix missing includes