Relax comparison to make test pass in float32