Fix heart class logic.