Add decorator for FMU