fix tree demo