Fix equals bug