Fix missing `include`