Fix build (missing headers)