fix test_forecast