Fixed float comparison in test.

Testing floats is always tricky.