Added the test for the previous changes.