fix forest and reshape test