Fix error in v1:test_river task