Updated an old junit import