Add two more unit tests for consistency