diff --git a/source/MainComponent.cpp b/source/MainComponent.cpp
index 7af4672..4a87147 100644
--- a/source/MainComponent.cpp
+++ b/source/MainComponent.cpp
@@ -892,7 +892,9 @@ const juce::String& MainComponent::getBaseDir()
 
 void MainComponent::run()
 {
+    trainButton->setButtonText (TRANS("Stop"));
 	m_pLayer->train(m_stack->trainingData(), this);
+    trainButton->setButtonText (TRANS("Train"));
 }
 
 bool MainComponent::onProgress(Rbm *pRbm, const Rbm::Status &status)