diff --git a/wireguard/wireguard.py b/wireguard/wireguard.py
index a634f13..48913d9 100644
--- a/wireguard/wireguard.py
+++ b/wireguard/wireguard.py
@@ -57,7 +57,7 @@ def generate_config(server_ip, server_port=51820):
                 if [ "$1" = "down" ]; then
                     sudo wg-quick down ~/.wg/server.conf >/dev/null 2>&1
                     echo "interface: server deleted"
-                fi
+                else
                     echo
                     sudo wg
                     echo
@@ -66,12 +66,13 @@ def generate_config(server_ip, server_port=51820):
                 if [ "$1" = "down" ]; then
                     echo "no interface: server"
                 else
-                sudo wg-quick up ~/.wg/server.conf >/dev/null 2>&1
-                echo
-                echo "interface: server created"
-                echo
-                sudo wg
-                echo
+                    sudo wg-quick up ~/.wg/server.conf >/dev/null 2>&1
+                    echo
+                    echo "interface: server created"
+                    echo
+                    sudo wg
+                    echo
+                fi
             fi
         '''))
         
@@ -105,7 +106,7 @@ def generate_config(server_ip, server_port=51820):
                 if [ "$1" = "down" ]; then
                     sudo wg-quick down ~/.wg/client.conf >/dev/null 2>&1
                     echo "interface: client deleted"
-                fi
+                else
                     echo
                     sudo wg
                     echo
@@ -114,12 +115,13 @@ def generate_config(server_ip, server_port=51820):
                 if [ "$1" = "down" ]; then
                     echo "no interface: client"
                 else
-                sudo wg-quick up ~/.wg/client.conf >/dev/null 2>&1
-                echo
-                echo "interface: client created"
-                echo
-                sudo wg
-                echo
+                    sudo wg-quick up ~/.wg/client.conf >/dev/null 2>&1
+                    echo
+                    echo "interface: client created"
+                    echo
+                    sudo wg
+                    echo
+                fi
             fi
         '''))