diff --git a/python/src/com/jetbrains/python/ui/IdeaDialog.java b/python/src/com/jetbrains/python/ui/IdeaDialog.java
index 241e5696d5c1..84ffd8fd9bd3 100644
--- a/python/src/com/jetbrains/python/ui/IdeaDialog.java
+++ b/python/src/com/jetbrains/python/ui/IdeaDialog.java
@@ -47,7 +47,7 @@ public abstract  class IdeaDialog extends DialogWrapper {
         field.getDocument().addDocumentListener(new MyDocumentListener());
     }
 
-    protected void addUpdater(JCheckBox check) {
+    protected void addUpdater(JToggleButton check) {
         check.addActionListener(new ActionListener() {
             public void actionPerformed(ActionEvent actionEvent) {
                 updateOkButton();