Turns out you need to wait till #end() is called for themes to work