Add missing include for infrastructure.h (#11)