[Test] Fix test compilation