diff --git a/source/MainComponent.cpp b/source/MainComponent.cpp
index 5c80082..cdace60 100644
--- a/source/MainComponent.cpp
+++ b/source/MainComponent.cpp
@@ -941,6 +941,9 @@ bool MainComponent::onProgress(Rbm *pRbm, const Rbm::Status &status)
 void MainComponent::updateControls()
 {
 	rbmDoRaoBlackwellToggleButton->setToggleState(m_pLayer->params().doRaoBlackwell, dontSendNotification);
+	rbmUseVisibleGaussianToggleButton->setToggleState(m_pLayer->params().doGaussianVisible, dontSendNotification);
+	rbmUseHiddenGaussianToggleButton->setToggleState(m_pLayer->params().doGaussianHidden, dontSendNotification);
+
 	rbmDoSampleVisibleToggleButton->setToggleState(m_pLayer->params().gibbsDoSampleVisible, dontSendNotification);
 	rbmDoSampleHidden->setToggleState(m_pLayer->params().gibbsDoSampleHidden, dontSendNotification);