Added correct Assert statements to unit tests.