Added a test for the new feature