From 928ea2f5cb7edf22d77ff3bc4bbca1b36c92b7cd Mon Sep 17 00:00:00 2001
From: Amjad Mashaal <me@amjad.io>
Date: Sat, 28 May 2016 02:18:49 +0200
Subject: [PATCH] Fixing minute problems with code

---
 certbot/main.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/certbot/main.py b/certbot/main.py
index 235d7c3ff..9c8849cfe 100644
--- a/certbot/main.py
+++ b/certbot/main.py
@@ -1,6 +1,7 @@
 """Certbot main entry point."""
 from __future__ import print_function
 import atexit
+import dialog
 import functools
 import logging.handlers
 import os
@@ -623,8 +624,7 @@ def _handle_exception(exc_type, exc_value, trace, config):
             # Here we're passing a client or ACME error out to the client at the shell
             # Tell the user a bit about what happened, without overwhelming
             # them with a full traceback
-            from dialog import DialogError
-            if issubclass(exc_type, DialogError):
+            if issubclass(exc_type, dialog.error):
                 err = exc_value.complete_message()
             else:
                 err = traceback.format_exception_only(exc_type, exc_value)[0]