diff --git a/tests/hitl/2_health.py b/tests/hitl/2_health.py
index d44633c6..d6c60273 100644
--- a/tests/hitl/2_health.py
+++ b/tests/hitl/2_health.py
@@ -38,10 +38,10 @@ def test_hw_type(p):
 def test_heartbeat(p, panda_jungle):
   panda_jungle.set_ignition(True)
   # TODO: add more cases here once the tests aren't super slow
-  p.set_safety_mode(mode=CarParams.SafetyModel.hyundai, param=HyundaiSafetyFlags.FLAG_HYUNDAI_LONG)
+  p.set_safety_mode(mode=CarParams.SafetyModel.hyundai, param=HyundaiSafetyFlags.LONG)
   p.send_heartbeat()
   assert p.health()['safety_mode'] == CarParams.SafetyModel.hyundai
-  assert p.health()['safety_param'] == HyundaiSafetyFlags.FLAG_HYUNDAI_LONG
+  assert p.health()['safety_param'] == HyundaiSafetyFlags.LONG
 
   # shouldn't do anything once we're in a car safety mode
   p.set_heartbeat_disabled()