I have a similar code in my theme and I figured out that I just have to add it before the last <input /> element.