Fix test
(cherry picked from commit 9c12118781e908b70240326dc6c49da112a35be0)