[misc] Improve tests