Correct Unit test