Fixes custom template function bug.