Make new unit test compatible with float32