trying to fix broken decorator