Add unit test for the internal state after removing controls