JUNIT test updates