Fix the equals method