Small tweak to utility function