fix(tests): to reflect recent changes

Signed-off-by: Leonardo de Moura <leonardo@microsoft.com>