Fix a bug. Add another test.

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