diff --git a/src/Mod/PartDesign/WizardShaft/WizardShaft.py b/src/Mod/PartDesign/WizardShaft/WizardShaft.py
index e08da93870..61ed019a02 100644
--- a/src/Mod/PartDesign/WizardShaft/WizardShaft.py
+++ b/src/Mod/PartDesign/WizardShaft/WizardShaft.py
@@ -88,7 +88,7 @@ class TaskWizardShaft:
         self.form.setLayout(layout)
         
         # Switch to feature window
-        mdi=QtGui.QApplication.activeWindow().findChild(QtGui.QMdiArea)
+        mdi=FreeCADGui.getMainWindow().findChild(QtGui.QMdiArea)
         cw.setActiveSubWindow(featureWindow)
         
     def slotAllx(self):