diff --git a/src/Insight/UserInterfaces/InsightWindow/Widgets/TargetWidgets/TargetPackageWidget.cpp b/src/Insight/UserInterfaces/InsightWindow/Widgets/TargetWidgets/TargetPackageWidget.cpp
index 032d747e..1cfce8ec 100644
--- a/src/Insight/UserInterfaces/InsightWindow/Widgets/TargetWidgets/TargetPackageWidget.cpp
+++ b/src/Insight/UserInterfaces/InsightWindow/Widgets/TargetWidgets/TargetPackageWidget.cpp
@@ -23,15 +23,11 @@ TargetPackageWidget::TargetPackageWidget(
 
     this->connect(
         &(this->insightWorker),
-        &InsightWorker::targetPinStatesUpdated,
+        &InsightWorker::targetIoPortsUpdated,
         this,
-        [this] (int variantId, const Targets::TargetPinStateMappingType& pinStatesByNumber) {
-            if (variantId == this->targetVariant.id) {
-                this->updatePinStates(pinStatesByNumber);
-
-                if (this->targetState == TargetState::STOPPED && !this->isEnabled()) {
-                    this->setDisabled(false);
-                }
+        [this] {
+            if (this->targetState == TargetState::STOPPED) {
+                this->refreshPinStates();
             }
         }
     );
diff --git a/src/Insight/UserInterfaces/InsightWindow/Widgets/TargetWidgets/TargetPackageWidget.hpp b/src/Insight/UserInterfaces/InsightWindow/Widgets/TargetWidgets/TargetPackageWidget.hpp
index 236996a5..c99f0366 100644
--- a/src/Insight/UserInterfaces/InsightWindow/Widgets/TargetWidgets/TargetPackageWidget.hpp
+++ b/src/Insight/UserInterfaces/InsightWindow/Widgets/TargetWidgets/TargetPackageWidget.hpp
@@ -33,5 +33,9 @@ namespace Bloom::Widgets::InsightTargetWidgets
     public:
         TargetPackageWidget(Targets::TargetVariant targetVariant, InsightWorker& insightWorker, QWidget* parent);
         virtual void refreshPinStates(std::optional<std::function<void(void)>> callback = std::nullopt);
+
+        virtual void setTargetState(Targets::TargetState targetState) {
+            this->targetState = targetState;
+        }
     };
 }