Fix missing headers