[Misc] Minor test improvement