diff --git a/Source/NeuropixCanvas.cpp b/Source/NeuropixCanvas.cpp
index 2ac6555..884efdb 100644
--- a/Source/NeuropixCanvas.cpp
+++ b/Source/NeuropixCanvas.cpp
@@ -241,7 +241,7 @@ void NeuropixCanvas::startAcquisition()
 {
     for (auto settingsInterface : settingsInterfaces)
     {
-        if (settingsInterface->dataSource != nullptr && settingsInterface->dataSource->isEnabled)
+        if (settingsInterface->dataSource != nullptr)
         {
             settingsInterface->startAcquisition();
         }
@@ -252,7 +252,7 @@ void NeuropixCanvas::stopAcquisition()
 {
     for (auto settingsInterface : settingsInterfaces)
     {
-        if (settingsInterface->dataSource != nullptr && settingsInterface->dataSource->isEnabled)
+        if (settingsInterface->dataSource != nullptr)
         {
             settingsInterface->stopAcquisition();
         }
diff --git a/Source/UI/NeuropixInterface.cpp b/Source/UI/NeuropixInterface.cpp
index 85cee3b..2920f18 100644
--- a/Source/UI/NeuropixInterface.cpp
+++ b/Source/UI/NeuropixInterface.cpp
@@ -1173,6 +1173,9 @@ void NeuropixInterface::startAcquisition()
     if (enableButton != nullptr)
         enableButton->setEnabled (enabledState);
 
+    if (probeEnableButton != nullptr)
+        probeEnableButton->setEnabled (enabledState);
+
     if (electrodeConfigurationComboBox != nullptr)
         electrodeConfigurationComboBox->setEnabled (enabledState);
 
@@ -1236,6 +1239,9 @@ void NeuropixInterface::stopAcquisition()
     if (enableButton != nullptr)
         enableButton->setEnabled (enabledState);
 
+    if (probeEnableButton != nullptr)
+        probeEnableButton->setEnabled (enabledState);
+
     if (electrodeConfigurationComboBox != nullptr)
         electrodeConfigurationComboBox->setEnabled (enabledState);