diff --git a/modules/swap/module.py b/modules/swap/module.py
index 20d9510..e810041 100644
--- a/modules/swap/module.py
+++ b/modules/swap/module.py
@@ -11,7 +11,7 @@ class SwapModule(AbstractModule):
 
     def run(self):
 
-        actions = ["Create/Resize", "Delete", "Get location", "Get swap size"]
+        actions = ["Create/Resize", "Delete", "Get swap location", "Get swap size"]
         menu = [
             {
                 'type': 'list',
@@ -21,13 +21,13 @@ class SwapModule(AbstractModule):
             }
         ]
         selected_action = prompt(menu)['action']
-        if selected_action is "Create/Resize":
+        if selected_action == "Create/Resize":
             self.create_resize_swap()
-        elif selected_action is "Delete":
+        elif selected_action == "Delete":
             self.delete_swap()
-        elif selected_action is "Get location":
+        elif selected_action == "Get swap location":
             self.swap_location_check()
-        elif selected_action is "Get swap size":
+        elif selected_action == "Get swap size":
             self.get_swap_size()
 
     def swap_location_check(self) -> None: