Added a test in unit test.