diff --git a/lua/neogen/configurations/python.lua b/lua/neogen/configurations/python.lua
index 39b88de..ae33e01 100644
--- a/lua/neogen/configurations/python.lua
+++ b/lua/neogen/configurations/python.lua
@@ -113,8 +113,7 @@ return {
             { nil, '""" $1 """', { no_results = true } },
             { "parameters", "\t%s: $1", { before_first_item = { "", "Args:" } } },
             { "attributes", "\t%s: $1", { before_first_item = { "", "Attributes: " } } },
-            { "return_statement", "", { before_first_item = { "", "Returns: " } } },
-            { nil, "" },
+            { "return_statement", "\t$1", { before_first_item = { "", "Returns: " } } },
             { nil, '"""' },
         },
         numpydoc = {
@@ -122,8 +121,7 @@ return {
             { nil, '""" $1 """', { no_results = true } },
             { "parameters", "%s: $1", { before_first_item = { "", "Parameters", "----------" } } },
             { "attributes", "%s: $1", { before_first_item = { "", "Attributes", "----------" } } },
-            { "return_statement", "", { before_first_item = { "", "Returns", "-------" } } },
-            { nil, "" },
+            { "return_statement", "$1", { before_first_item = { "", "Returns", "-------" } } },
             { nil, '"""' },
         },
     },