Correct Junit tests