[Test] Fixed failing test