Fixed the == bug correctly