Correct unittest