Fix test for _forecast partial