diff --git a/lua/neogen/configurations/python.lua b/lua/neogen/configurations/python.lua
index 8849900..9aad0b3 100644
--- a/lua/neogen/configurations/python.lua
+++ b/lua/neogen/configurations/python.lua
@@ -101,6 +101,9 @@ return {
                         if res["return_type"] then
                             res["return_statement"] = nil
                             res["anonymous_return"] = nil
+                            if res["return_type"][1] == "None" then
+                                res["return_type"] = nil
+                            end
                         end
 
                         -- Remove reference to "self" if any