small test improvement