Tweak UltiSnips callback function.