Fix unit tests now that everything is Float32