Update assertEquals method