X-Git-Url: https://git.saurik.com/wxWidgets.git/blobdiff_plain/95bfd958bda8955ac81277c78e5627c926154f39..628eae0bcd7916c57847b7eab739d4cc32534ccf:/wxPython/demo/ToggleButton.py

diff --git a/wxPython/demo/ToggleButton.py b/wxPython/demo/ToggleButton.py
index 43a7d0a370..f03dd7f4b7 100644
--- a/wxPython/demo/ToggleButton.py
+++ b/wxPython/demo/ToggleButton.py
@@ -2,10 +2,7 @@
 import  wx
 
 haveToggleBtn = 1
-
-try:
-    wx.ToggleButton
-except NameError:
+if wx.Platform == "__WXX11__":
     haveToggleBtn = 0
 
 #----------------------------------------------------------------------
@@ -37,10 +34,10 @@ def runTest(frame, nb, log):
         win = TestPanel(nb, log)
         return win
     else:
-        dlg = wx.MessageDialog(frame, 'wx.ToggleButton is not available on this platform.',
-                          'Sorry', wx.OK | wx.ICON_INFORMATION)
-        dlg.ShowModal()
-        dlg.Destroy()
+        from Main import MessagePanel
+        win = MessagePanel(nb, 'wx.ToggleButton is not available on this platform.',
+                           'Sorry', wx.ICON_WARNING)
+        return win
 
 
 #----------------------------------------------------------------------
@@ -60,4 +57,4 @@ This class is only available under wxMSW and wxGTK currently.
 if __name__ == '__main__':
     import sys,os
     import run
-    run.main(['', os.path.basename(sys.argv[0])])
+    run.main(['', os.path.basename(sys.argv[0])] + sys.argv[1:])