change approach to junit