Minor test improvement