found the unittest and added test for the new feature.