[MIE-1633] Fixed tests