diff --git a/mlir/include/mlir/Dialect/Linalg/ComprehensiveBufferize/ComprehensiveBufferize.h b/mlir/include/mlir/Dialect/Linalg/ComprehensiveBufferize/ComprehensiveBufferize.h
index 295fd70cb156..93a697fbeb39 100644
--- a/mlir/include/mlir/Dialect/Linalg/ComprehensiveBufferize/ComprehensiveBufferize.h
+++ b/mlir/include/mlir/Dialect/Linalg/ComprehensiveBufferize/ComprehensiveBufferize.h
@@ -17,7 +17,7 @@ namespace linalg {
 namespace comprehensive_bufferize {
 
 struct BufferizationOptions;
-struct BufferizationState;
+class BufferizationState;
 struct PostAnalysisStep;
 
 /// Bufferize the given operation. Reuses an existing BufferizationState object.