[misc] Updated tests