diff --git a/src/CppParser/Parser.cpp b/src/CppParser/Parser.cpp
index 22e3a40c..91a264f1 100644
--- a/src/CppParser/Parser.cpp
+++ b/src/CppParser/Parser.cpp
@@ -1195,8 +1195,8 @@ TranslationUnit* Parser::GetTranslationUnit(clang::SourceLocation Loc,
 
     auto Unit = Lib->FindOrCreateModule(File);
 
-    if (Unit->OriginalPtr == nullptr)
-        Unit->OriginalPtr = (void*) SM.getFileEntryForID(SM.getFileID(Loc));
+    Unit->OriginalPtr = (void*) Unit;
+    assert(Unit->OriginalPtr != nullptr);
 
     if (LocKind != SourceLocationKind::Invalid)
         Unit->IsSystemHeader = SM.isInSystemHeader(Loc);