Fix floating point comparison issue