Fix wrong decorator