[Test] Fix test