1 files changed, 3 insertions, 0 deletions
diff --git a/clang/lib/AST/DeclBase.cpp b/clang/lib/AST/DeclBase.cpp
index d872eae0845..3d92331792d 100644
--- a/clang/lib/AST/DeclBase.cpp
+++ b/clang/lib/AST/DeclBase.cpp
@@ -158,6 +158,9 @@ void Decl::setLexicalDeclContext(DeclContext *DC) {
 }
 
 TranslationUnitDecl *Decl::getTranslationUnitDecl() {
+  if (TranslationUnitDecl *TUD = dyn_cast<TranslationUnitDecl>(this))
+    return TUD;
+
   DeclContext *DC = getDeclContext();
   assert(DC && "This decl is not contained in a translation unit!");