Fix failing equality test