Added test for new util
Bug fix