I'm trying to get all the flow runs in Running state with client.read_flow_runs() but got error when initialize the FlowRunFilterState object. I couldn't find any example to do this from discourse, does anyone know what I did wrong?

import asyncio
from prefect.client import get_client
from prefect.orion.schemas.states import StateType
from prefect.orion.schemas.filters import FlowRunFilter, FlowRunFilterState, FlowRunFilterStateType

async def main():
    client = get_client()
    state_type_filter = FlowRunFilterStateType(any_=[StateType.RUNNING])
    state_filter = FlowRunFilterState(state_type_filter)
    flow_filter = FlowRunFilter(state_filter)
    flow_runs = await client.read_flow_runs(flow_filter)
    print(flow_runs)

if __name__ == '__main__':
    asyncio.run(main())

Traceback (most recent call last):
  File "get_client_context.py", line 21, in <module>
    asyncio.run(main())
  File "/usr/lib/python3.8/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/usr/lib/python3.8/asyncio/base_events.py", line 616, in run_until_complete
    return future.result()
  File "get_client_context.py", line 9, in main
    state_filter = FlowRunFilterState(state_type_filter)
  File "pydantic/main.py", line 333, in pydantic.main.BaseModel.__init__
TypeError: __init__() takes exactly 1 positional argument (2 given)