1 files changed, 1 insertions, 0 deletions
diff --git a/main.go b/main.go
index d1ccf4a..a934a22 100644
--- a/main.go
+++ b/main.go
@@ -23,6 +23,7 @@ func main() {
 			choice = 1
 		case "edit": // Import table
 			choice = 3
+			table = os.Args[2]
 		case "run": // Run query
 			choice = 2
 			query = strings.Join(os.Args[2:], " ")