[CI] Fixed tests