diff --git a/Lib/idlelib/PyShell.py b/Lib/idlelib/PyShell.py
index cde389ad009..d8a5cd24331 100644
--- a/Lib/idlelib/PyShell.py
+++ b/Lib/idlelib/PyShell.py
@@ -61,6 +61,7 @@ def idle_showwarning(message, category, filename, lineno,
                                               lineno, line=line))
         except OSError:
             pass  ## file (probably __stderr__) is invalid, warning dropped.
+    warnings.showwarning = idle_showwarning
     def idle_formatwarning(message, category, filename, lineno, line=None):
         """Format warnings the IDLE way"""
         s = "\nWarning (from warnings module):\n"
@@ -72,6 +73,7 @@ def idle_formatwarning(message, category, filename, lineno, line=None):
             s += "    %s\n" % line
         s += "%s: %s\n>>> " % (category.__name__, message)
         return s
+    warnings.formatwarning = idle_formatwarning
 
 def extended_linecache_checkcache(filename=None,
                                   orig_checkcache=linecache.checkcache):
@@ -1423,9 +1425,6 @@ def close(self):
 def main():
     global flist, root, use_subprocess
 
-    warnings.showwarning = idle_showwarning
-    warnings.formatwarning = idle_formatwarning
-
     use_subprocess = True
     enable_shell = False
     enable_edit = False
diff --git a/Lib/idlelib/run.py b/Lib/idlelib/run.py
index 74c1b8947a0..15f44720dd7 100644
--- a/Lib/idlelib/run.py
+++ b/Lib/idlelib/run.py
@@ -40,6 +40,7 @@ def idle_formatwarning_subproc(message, category, filename, lineno,
             s += "    %s\n" % line
         s += "%s: %s\n" % (category.__name__, message)
         return s
+    warnings.formatwarning = idle_formatwarning_subproc
 
 
 tcl = tkinter.Tcl()
@@ -81,9 +82,6 @@ def main(del_exitfunc=False):
     global exit_now
     global quitting
     global no_exitfunc
-
-    warnings.formatwarning = idle_formatwarning_subproc
-
     no_exitfunc = del_exitfunc
     #time.sleep(15) # test subprocess not responding
     try: