[misc] Fix tests