fix decorator demo code