diff --git a/Train3DMASCDialog.ui b/Train3DMASCDialog.ui
index 3d32dd7..19d5193 100644
--- a/Train3DMASCDialog.ui
+++ b/Train3DMASCDialog.ui
@@ -228,6 +228,13 @@
         </column>
        </widget>
       </item>
+      <item row="3" column="0">
+       <widget class="QToolButton" name="exportToolButton">
+        <property name="text">
+         <string>export</string>
+        </property>
+       </widget>
+      </item>
       <item row="1" column="1">
        <widget class="QTableWidget" name="tableWidgetScales">
         <column>
@@ -237,26 +244,6 @@
         </column>
        </widget>
       </item>
-      <item row="2" column="1">
-       <spacer name="verticalSpacer">
-        <property name="orientation">
-         <enum>Qt::Vertical</enum>
-        </property>
-        <property name="sizeHint" stdset="0">
-         <size>
-          <width>20</width>
-          <height>190</height>
-         </size>
-        </property>
-       </spacer>
-      </item>
-      <item row="3" column="0">
-       <widget class="QToolButton" name="exportToolButton">
-        <property name="text">
-         <string>export</string>
-        </property>
-       </widget>
-      </item>
      </layout>
     </widget>
    </item>
diff --git a/q3DMASC.cpp b/q3DMASC.cpp
index 306e655..74bf945 100644
--- a/q3DMASC.cpp
+++ b/q3DMASC.cpp
@@ -382,6 +382,7 @@ void q3DMASCPlugin::doTrainAction()
 	}
 	for(double scale : scales)
 		trainDlg.addScale(scale, true);
+	trainDlg.connectScaleSelectionToFeatureSelection();
 
 	std::vector<FeatureSelection> originalFeaturesTest;
 	if (testCloud && needTestSuite)
diff --git a/qTrain3DMASCDialog.cpp b/qTrain3DMASCDialog.cpp
index c77e55d..75f19ab 100644
--- a/qTrain3DMASCDialog.cpp
+++ b/qTrain3DMASCDialog.cpp
@@ -75,6 +75,24 @@ int Train3DMASCDialog::addScale(double scale, bool isChecked/*=true*/)
 	return index;
 }
 
+void Train3DMASCDialog::scaleStateChanged(QTableWidgetItem* item)
+{
+	// if the scale is not checked, remove automatically features based on this scale
+	QString scale = "_SC" + item->text() + "_";
+	for (int row = 0; row < tableWidget->rowCount(); row++)
+	{
+		QTableWidgetItem* nameItem = tableWidget->item(row, 0);
+		QString name = nameItem->text();
+		if (name.contains(scale))
+			nameItem->setCheckState(item->checkState());
+	}
+}
+
+void Train3DMASCDialog::connectScaleSelectionToFeatureSelection()
+{
+	connect(tableWidgetScales, &QTableWidget::itemChanged, this, &Train3DMASCDialog::scaleStateChanged);
+}
+
 void Train3DMASCDialog::setResultText(QString text)
 {
 	resultLabel->setText(text);
diff --git a/qTrain3DMASCDialog.h b/qTrain3DMASCDialog.h
index 153eb3b..56f5bf9 100644
--- a/qTrain3DMASCDialog.h
+++ b/qTrain3DMASCDialog.h
@@ -39,6 +39,8 @@ public:
 	**/
 	int addFeature(QString name, float importance, bool isChecked = true);
 	int addScale(double scale, bool isChecked = true);
+	void scaleStateChanged(QTableWidgetItem* item);
+	void connectScaleSelectionToFeatureSelection();
 	void setResultText(QString text);
 	void setFirstRunDone();
 	inline void setClassifierSaved() { classifierSaved = true; saveRequested = false; }