Fix building (missing header)