Correct Unit Test