[misc] Fixed failing tests