Updated util function