Tweak template code