Added missing cases to the tests.