Final junit test corrected