From 059a7c6dd31f036771b3ce94cd5f064696d64229 Mon Sep 17 00:00:00 2001
From: Sidharth Pal <sidharth.pal1992@gmail.com>
Date: Fri, 5 Nov 2021 19:54:43 +0100
Subject: [PATCH] Update utils.py

---
 doltcli/utils.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/doltcli/utils.py b/doltcli/utils.py
index 409ddbe..cb231c9 100644
--- a/doltcli/utils.py
+++ b/doltcli/utils.py
@@ -254,6 +254,8 @@ def _get_import_mode_and_flags(
     import_modes = IMPORT_MODES_TO_FLAGS.keys()
     if import_mode and import_mode not in import_modes:
         raise ValueError(f"update_mode must be one of: {import_modes}")
+    elif import_mode:
+        return import_mode
     else:
         if table in [table.name for table in dolt.ls()]:
             logger.info(f'No import mode specified, table exists, using "{UPDATE}"')