diff --git a/lib/Dialect/Substrait/IR/Substrait.cpp b/lib/Dialect/Substrait/IR/Substrait.cpp
index c91e574..c3f9880 100644
--- a/lib/Dialect/Substrait/IR/Substrait.cpp
+++ b/lib/Dialect/Substrait/IR/Substrait.cpp
@@ -436,8 +436,11 @@ LogicalResult ProjectOp::verifyRegions() {
 } // namespace mlir
 
 //===----------------------------------------------------------------------===//
-// Substrait types
+// Substrait types and attributes
 //===----------------------------------------------------------------------===//
 
 #define GET_TYPEDEF_CLASSES
 #include "substrait-mlir/Dialect/Substrait/IR/SubstraitOpsTypes.cpp.inc"
+
+#define GET_ATTRDEF_CLASSES
+#include "substrait-mlir/Dialect/Substrait/IR/SubstraitOpsAttrs.cpp.inc"
\ No newline at end of file