diff --git a/src/main/java/org/jetbrains/plugins/template/TreeTransferHandler.java b/src/main/java/org/jetbrains/plugins/template/TreeTransferHandler.java
index 810067c..afca398 100644
--- a/src/main/java/org/jetbrains/plugins/template/TreeTransferHandler.java
+++ b/src/main/java/org/jetbrains/plugins/template/TreeTransferHandler.java
@@ -125,15 +125,15 @@ private DefaultMutableTreeNode copy(DefaultMutableTreeNode node, HashSet<TreeNod
 
     protected void exportDone(JComponent source, Transferable data, int action) {
         if ((action & MOVE) == MOVE) {
+            // Set the "Apply" Button in the settings modal to be enabled
+            configurable.setApplyToTrue();
+
             JTree tree = (JTree) source;
             DefaultTreeModel model = (DefaultTreeModel) tree.getModel();
             // Remove nodes saved in nodesToRemove in createTransferable.
             for (int i = 0; i < nodesToRemove.length; i++) {
                 model.removeNodeFromParent(nodesToRemove[i]);
             }
-
-            // Set the "Apply" Button in the settings modal to be enabled
-            configurable.setApplyToTrue();
         }
     }