corrected unit tests to new data