diff --git a/nvim/UltiSnips/python.snippets b/nvim/UltiSnips/python.snippets
index 573e2fc..38da3c3 100644
--- a/nvim/UltiSnips/python.snippets
+++ b/nvim/UltiSnips/python.snippets
@@ -7,12 +7,12 @@ snippet date
 endsnippet
 
 snippet ifmain
-def main():
-    """$1"""
-		return
+def $1():
+    """$2"""
+    return
 
 if __name__ == '__main__':
-    main()
+    $1()
 endsnippet
 
 snippet logger