FIX: Decorate tests