Fix missing include breaking compilation