diff --git a/src/Extensions/Hardware.bonsai b/src/Extensions/Hardware.bonsai
index c1730e2..70a5223 100644
--- a/src/Extensions/Hardware.bonsai
+++ b/src/Extensions/Hardware.bonsai
@@ -196,7 +196,7 @@
RigSchema
- HarpSniffDetector
+ HarpEnvironmentSensor
diff --git a/src/aind_behavior_dynamic_foraging/data_qc/suite.py b/src/aind_behavior_dynamic_foraging/data_qc/suite.py
index b0602f7..a0033ae 100644
--- a/src/aind_behavior_dynamic_foraging/data_qc/suite.py
+++ b/src/aind_behavior_dynamic_foraging/data_qc/suite.py
@@ -68,12 +68,14 @@ def make_qc_runner(dataset: contract.Dataset) -> qc.Runner:
"HarpEnvironmentSensor",
)
- _runner.add_suite(
- qc.harp.HarpLicketySplitTestSuite(dataset["Behavior"]["HarpLickometerRight"]), "HarpLickometerRight"
- )
- _runner.add_suite(
- qc.harp.HarpLicketySplitTestSuite(dataset["Behavior"]["HarpLickometerLeft"]), "HarpLickometerLeft"
- )
+ if rig.harp_lickometer_right is not None:
+ _runner.add_suite(
+ qc.harp.HarpLicketySplitTestSuite(dataset["Behavior"]["HarpLickometerRight"]), "HarpLickometerRight"
+ )
+ if rig.harp_lickometer_left is not None:
+ _runner.add_suite(
+ qc.harp.HarpLicketySplitTestSuite(dataset["Behavior"]["HarpLickometerLeft"]), "HarpLickometerLeft"
+ )
# Add camera qc
for camera in dataset["BehaviorVideos"]: