diff --git a/op-finder-lib/src/OperationFinderAstVisitor.cpp b/op-finder-lib/src/OperationFinderAstVisitor.cpp
index 39833c8..2421c05 100644
--- a/op-finder-lib/src/OperationFinderAstVisitor.cpp
+++ b/op-finder-lib/src/OperationFinderAstVisitor.cpp
@@ -120,7 +120,8 @@ bool OperationFinderAstVisitor::dataTraverseStmtPost(clang::Stmt* stmt)
 
 clang::Stmt* OperationFinderAstVisitor::_isBranchEntry(clang::Stmt* stmt)
 {
-  if (auto* loop = clang::dyn_cast<clang::ForStmt>(stmt))
+  if (auto* loop = clang::dyn_cast<clang::ForStmt>(stmt);
+      loop && loop->getInit())
   {
     _loop_header.init = loop->getInit();