diff --git a/board/pyro/battery.c b/board/pyro/battery.c
index 87aacfec72..1b816cadd1 100644
--- a/board/pyro/battery.c
+++ b/board/pyro/battery.c
@@ -232,6 +232,8 @@ int charger_profile_override(struct charge_state_data *curr)
 	int batt_voltage;
 	static int prev_batt_voltage;
 
+	charger_discharge_on_ac(!(curr->batt.flags & BATT_FLAG_WANT_CHARGE));
+
 	/*
 	 * Determine temperature range. The five ranges are:
 	 *   < 10C
diff --git a/board/snappy/battery.c b/board/snappy/battery.c
index ec7543a78d..4abdff5534 100644
--- a/board/snappy/battery.c
+++ b/board/snappy/battery.c
@@ -160,6 +160,8 @@ int charger_profile_override(struct charge_state_data *curr)
 	int batt_voltage;
 	static int prev_batt_voltage;
 
+	charger_discharge_on_ac(!(curr->batt.flags & BATT_FLAG_WANT_CHARGE));
+
 	/*
 	 * Determine temperature range. The five ranges are:
 	 *   < 10C