From dc3c7eb4eadd44a6b3371af23e6089589306f9ee Mon Sep 17 00:00:00 2001
From: Daniel Mathiot <d.danymat@gmail.com>
Date: Tue, 17 Aug 2021 21:46:09 +0200
Subject: [PATCH] Add query for return statement in an if block

---
 lua/neogen.lua | 1 +
 1 file changed, 1 insertion(+)

diff --git a/lua/neogen.lua b/lua/neogen.lua
index 8e52492..6d3f3af 100644
--- a/lua/neogen.lua
+++ b/lua/neogen.lua
@@ -8,6 +8,7 @@ M.generate = function ()
     (function (parameters) @params)
     (function (return_statement) @return)
     (function_definition (parameters) @params)
+    (function_definition (if_statement (return_statement) @return))
     ]]
 
     -- Try to find the upper function