diff --git a/tools/quantification/src/block_desc_local.h b/tools/quantification/src/block_desc_local.h
index 41c2dc0abbdf8bb006f4152674e92dd1f7d01500..a28d551071d6d309c67a6b11cda0962d6922dd45 100644
--- a/tools/quantification/src/block_desc_local.h
+++ b/tools/quantification/src/block_desc_local.h
@@ -20,6 +20,7 @@ limitations under the License. */
 #define TOOLS_QUANTIFICATION_SRC_BLOCK_DESC_LOCAL_H_
 
 #include <vector>
+#include <memory>
 #include "src/var_desc.h"
 
 class BlockDesc {