diff --git a/tcutils/loghelper.py b/tcutils/loghelper.py
index 5bebb21..7011cef 100644
--- a/tcutils/loghelper.py
+++ b/tcutils/loghelper.py
@@ -71,6 +71,7 @@
 def setup_logger(**kwargs):
     level = kwargs.get('log_level', 'info')
     level_str = dict(info=logging.INFO, debug=logging.DEBUG,
-                     error=logging.ERROR)
+                     error=logging.ERROR, warning=logging.WARNING,
+                     critical=logging.CRITICAL)
     logging.getLogger().setLevel(level_str.get(level, logging.INFO))
     _enable_pretty_logging(**kwargs)