BUG: Add the assertion in the test.