Fix missing decorator