Use async_function decorator properly