Finish tidying up ball tests