Add some missing decorator