X-Git-Url: https://git.saurik.com/wxWidgets.git/blobdiff_plain/8eca4fef106b8327e2e55636de3f68a511a4c392..3cebfef17b54f9dacc386f090827b8badb2773d2:/wxPython/demo/run.py?ds=inline

diff --git a/wxPython/demo/run.py b/wxPython/demo/run.py
index fb9b70a80f..a267dfd089 100755
--- a/wxPython/demo/run.py
+++ b/wxPython/demo/run.py
@@ -64,6 +64,7 @@ class RunDemoApp(wx.App):
         menuBar.Append(menu, "&File")
 
         ns = {}
+        ns['wx'] = wx
         ns['app'] = self
         ns['module'] = self.demoModule
         ns['frame'] = frame
@@ -92,9 +93,7 @@ class RunDemoApp(wx.App):
                 frect = frame.otherWin.GetRect()
                 p = wx.Panel(frame, -1)
                 b = wx.Button(p, -1, " Exit ", (10,10))
-                p.Fit()
-                frame.SetClientSize(p.GetSize())
-                #frame.SetSize((200, 100))
+                wx.CallAfter(frame.SetClientSize, (200, 100))
                 frame.Bind(wx.EVT_BUTTON, self.OnButton, b)
             else:
                 # It was probably a dialog or something that is already
@@ -119,7 +118,8 @@ class RunDemoApp(wx.App):
             # Hook the close event of the test window so that we close
             # the shell at the same time
             def CloseShell(evt):
-                shell.Close()
+                if shell:
+                    shell.Close()
                 evt.Skip()
             frame.Bind(wx.EVT_CLOSE, CloseShell)