Correct unit tests