Use theme callback insead.
Minor fix.