diff --git a/lua/neogen/configurations/c.lua b/lua/neogen/configurations/c.lua
index 0f518a4..28b7f9c 100644
--- a/lua/neogen/configurations/c.lua
+++ b/lua/neogen/configurations/c.lua
@@ -50,11 +50,11 @@ local c_function_extractor = function(node)
     end
 
     local has_return_statement = function()
-        -- function implementation
         if res.return_statement then
+            -- function implementation
             return res.return_statement
-            -- function prototype
         elseif res.function_declarator and res.primitive_type and res.primitive_type[1] ~= "void" then
+            -- function prototype
             return res.primitive_type
         end
         -- not found