[misc] Fixed tests