slight test cleanup