Correctly updated unit tests

Forgot to run the unit tests before checking in !!!!!