small test cleanup