Fix deprecated junit import