small fix to improve logger output

1 files changed, 3 insertions, 0 deletions

diff --git a/datamaps/main.py b/datamaps/main.py
index a36b468..6a7a018 100644
--- a/datamaps/main.py
+++ b/datamaps/main.py
@@ -28,6 +28,7 @@ from datamaps import __version__
 from engine.adapters import cli as engine_cli
 from engine.config import Config as engine_config
 from engine.exceptions import (MalFormedCSVHeaderException,
+                               NoApplicableSheetsInTemplateFiles,
                                RemoveFileWithNoSheetRequiredByDatamap)
 
 logging.basicConfig(level=logging.INFO, format="%(asctime)s: %(levelname)s - %(message)s", datefmt='%d-%b-%y %H:%M:%S')
@@ -112,6 +113,8 @@ def templates(to_master):
         except RuntimeError as e:
             logger.critical(e)
             logger.critical("Not completing import process.")
+        except NoApplicableSheetsInTemplateFiles as e:
+            logger.critical("Not completing import process.")
 
     else:
         click.secho("Not implemented yet. Try --to-master/-m flag")