fix decorator -- NameError