Add a simple test for the new cases.