diff --git a/lib/TableGen/GenDialect.cpp b/lib/TableGen/GenDialect.cpp
index 3436c75..f533b13 100644
--- a/lib/TableGen/GenDialect.cpp
+++ b/lib/TableGen/GenDialect.cpp
@@ -237,6 +237,7 @@ void llvm_dialects::genDialectDefs(raw_ostream& out, RecordKeeper& records) {
 #include "llvm/ADT/SmallVector.h"
 #include "llvm/ADT/StringExtras.h"
 #include "llvm/IR/InstrTypes.h"
+#include "llvm/IR/Module.h"
 )";
 
   if (!noMemoryEffects()) {