Fix == comparison bug