Add missing header to catch correctness issues.