Correct unit test