Fixing minute problems with code

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]